updates

2019-11-20 13:40:24 -08:00
parent bb209111c4
commit 3a4ed8b3ab
2 changed files with 2 additions and 2 deletions
@@ -44,7 +44,7 @@ def load_classes(path):
 def labels_to_class_weights(labels, nc=80):
    # Get class weights (inverse frequency) from training labels
    if labels[0] is None:  # no labels loaded
-        return None
+        return torch.Tensor()

    labels = np.concatenate(labels, 0)  # labels.shape = (866643, 5) for COCO
    classes = labels[:, 0].astype(np.int)  # labels = [class xywh]