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():
|