updates

test.py

@@ -24,6 +24,10 @@ def test(cfg,
         device = torch_utils.select_device(opt.device, batch_size=batch_size)
         verbose = True
 
+        # Remove previous
+        for f in glob.glob('test_batch*.jpg'):
+            os.remove(f)
+
         # Initialize model
         model = Darknet(cfg, img_size).to(device)
 
@@ -36,7 +40,7 @@ def test(cfg,
 
         if torch.cuda.device_count() > 1:
             model = nn.DataParallel(model)
-    else:
+    else:  # called by train.py
         device = next(model.parameters()).device  # get model device
         verbose = False