now videodemo works without weights in folder

VCasecnikovs · Jul 20, 2020 · c8cc941 · c8cc941
1 parent 10251e9
commit c8cc941
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/video_demo.py b/video_demo.py
@@ -92,7 +92,7 @@
 threshold = 0.2
 iou_threshold = 0.2
 
-m = YOLOv4(weights_path="weights/yolov4.pth")
+m = YOLOv4(pretrained=True, sam=False, eca=False)
 m.requires_grad_(False)
 m.eval()