Update detector.py
Browse files- detector.py +12 -0
detector.py
CHANGED
@@ -394,6 +394,18 @@ def download_sample_images():
|
|
394 |
"https://github.com/Abs6187/Helmet-Detection/blob/main/Sample-Image-5.jpg?raw=true",
|
395 |
"sample_5.jpg",
|
396 |
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
397 |
|
398 |
|
399 |
def get_ocr_status():
|
|
|
394 |
"https://github.com/Abs6187/Helmet-Detection/blob/main/Sample-Image-5.jpg?raw=true",
|
395 |
"sample_5.jpg",
|
396 |
)
|
397 |
+
torch.hub.download_url_to_file(
|
398 |
+
"https://github.com/Abs6187/Helmet-Detection/blob/main/Sample-Image-6.jpg?raw=true",
|
399 |
+
"sample_6.jpg",
|
400 |
+
)
|
401 |
+
torch.hub.download_url_to_file(
|
402 |
+
"https://github.com/Abs6187/Helmet-Detection/blob/main/Sample-Image-7.jpg?raw=true",
|
403 |
+
"sample_7.jpg",
|
404 |
+
)
|
405 |
+
torch.hub.download_url_to_file(
|
406 |
+
"https://github.com/Abs6187/Helmet-Detection/blob/main/Sample-Image-8.jpg?raw=true",
|
407 |
+
"sample_8.jpg",
|
408 |
+
)
|
409 |
|
410 |
|
411 |
def get_ocr_status():
|