This commit is contained in:
Glenn Jocher 2019-12-03 13:34:22 -08:00
parent 896fd6d025
commit 0dd0fa7938
1 changed files with 2 additions and 1 deletions

View File

@ -449,7 +449,8 @@ def build_targets(model, targets):
# Class
tcls.append(c)
if c.shape[0]: # if any targets
assert c.max() <= model.nc, 'Target classes exceed model classes'
assert c.max() <= model.nc, 'Model accepts %g classes labeled from 0-%g, however you supplied a label %g. \
See https://github.com/ultralytics/yolov3/wiki/Train-Custom-Data' % (model.nc, model.nc - 1, c.max())
return tcls, tbox, indices, av