Abs6187 commited on
Commit
d2b4b74
·
verified ·
1 Parent(s): 736b620

Update detector.py

Browse files
Files changed (1) hide show
  1. 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():