提交 07493a71 authored 作者: Glenn Jocher's avatar Glenn Jocher

update train.py class count assertion #424

上级 912c06a6
...@@ -164,7 +164,7 @@ def train(hyp): ...@@ -164,7 +164,7 @@ def train(hyp):
hyp=hyp, augment=True, cache=opt.cache_images, rect=opt.rect) hyp=hyp, augment=True, cache=opt.cache_images, rect=opt.rect)
mlc = np.concatenate(dataset.labels, 0)[:, 0].max() # max label class mlc = np.concatenate(dataset.labels, 0)[:, 0].max() # max label class
nb = len(dataloader) # number of batches nb = len(dataloader) # number of batches
assert mlc < nc, 'Label class %g exceeds nc=%g in %s. Possible class labels are 0-%g' % (mlc, nc, opt.cfg, nc - 1) assert mlc < nc, 'Label class %g exceeds nc=%g in %s. Possible class labels are 0-%g' % (mlc, nc, opt.data, nc - 1)
# Testloader # Testloader
testloader = create_dataloader(test_path, imgsz_test, batch_size, gs, opt, testloader = create_dataloader(test_path, imgsz_test, batch_size, gs, opt,
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论