updates

2020-01-15 10:22:59 -08:00
parent 5831d2d6ba
commit 53e3d55a1e
2 changed files with 2 additions and 2 deletions
@@ -830,7 +830,7 @@ def print_mutation(hyp, results, bucket=''):

 def apply_classifier(x, model, img, im0):
    # applies a second stage classifier to yolo outputs
-
+    im0 = [im0] if isinstance(im0, np.ndarray) else im0
    for i, d in enumerate(x):  # per image
        if d is not None and len(d):
            d = d.clone()