fix : local로 돌릴 수 있도록 함

Capic2024 · May 14, 2024 · 7daa148 · 7daa148
1 parent c2125f1
commit 7daa148
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/__pycache__/mosaic.cpython-311.pyc b/__pycache__/mosaic.cpython-311.pyc
diff --git a/mosaic.py b/mosaic.py
@@ -27,7 +27,7 @@ def mosaic(video_path, image_paths):
     face_count = 0
     current_frame_count = 0
 
-    model = torch.hub.load('ultralytics/yolov5', 'custom', path='macWideface.pt', force_reload=True)
+    model = torch.hub.load('./yolov5', 'custom', path='macWideface.pt', force_reload=True, source='local')
 
     embedding_list = []
     for image_path in image_paths: