updates

2019-04-12 14:58:19 +02:00
parent 5ea92e7ee2
commit 50df252c4b
2 changed files with 3 additions and 4 deletions
@@ -321,7 +321,7 @@ def build_targets(model, targets):

        # Class
        tcls.append(c)
-        if nt:
+        if c.shape[0]:
            assert c.max() <= layer.nC, 'Target classes exceed model classes'

    return txy, twh, tcls, indices