Update detector.py
Browse files- detector.py +5 -5
detector.py
CHANGED
@@ -375,23 +375,23 @@ def yolov8_detect(
|
|
375 |
def download_sample_images():
|
376 |
"""Download sample images for testing."""
|
377 |
torch.hub.download_url_to_file(
|
378 |
-
"https://github.com/
|
379 |
"sample_1.jpg",
|
380 |
)
|
381 |
torch.hub.download_url_to_file(
|
382 |
-
"https://github.com/
|
383 |
"sample_2.jpg",
|
384 |
)
|
385 |
torch.hub.download_url_to_file(
|
386 |
-
"https://github.com/
|
387 |
"sample_3.jpg",
|
388 |
)
|
389 |
torch.hub.download_url_to_file(
|
390 |
-
"https://github.com/
|
391 |
"sample_4.jpg",
|
392 |
)
|
393 |
torch.hub.download_url_to_file(
|
394 |
-
"https://github.com/
|
395 |
"sample_5.jpg",
|
396 |
)
|
397 |
|
|
|
375 |
def download_sample_images():
|
376 |
"""Download sample images for testing."""
|
377 |
torch.hub.download_url_to_file(
|
378 |
+
"https://github.com/Abs6187/Helmet-Detection/blob/main/Sample-Image-1.jpg?raw=true",
|
379 |
"sample_1.jpg",
|
380 |
)
|
381 |
torch.hub.download_url_to_file(
|
382 |
+
"https://github.com/Abs6187/Helmet-Detection/blob/main/Sample-Image-2.jpg?raw=true",
|
383 |
"sample_2.jpg",
|
384 |
)
|
385 |
torch.hub.download_url_to_file(
|
386 |
+
"https://github.com/Abs6187/Helmet-Detection/blob/main/Sample-Image-3.jpg?raw=true",
|
387 |
"sample_3.jpg",
|
388 |
)
|
389 |
torch.hub.download_url_to_file(
|
390 |
+
"https://github.com/Abs6187/Helmet-Detection/blob/main/Sample-Image-4.jpg?raw=true",
|
391 |
"sample_4.jpg",
|
392 |
)
|
393 |
torch.hub.download_url_to_file(
|
394 |
+
"https://github.com/Abs6187/Helmet-Detection/blob/main/Sample-Image-5.jpg?raw=true",
|
395 |
"sample_5.jpg",
|
396 |
)
|
397 |
|