Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updates

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
utils/utils.py
View File

@ -449,7 +449,8 @@ def build_targets(model, targets):
# Class # Class
tcls.append(c) tcls.append(c)
if c.shape[0]: # if any targets 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 return tcls, tbox, indices, av