Upload app.py

app.py

@@ -7,50 +7,49 @@ import random
 from ultralytics import YOLO
 
 file_urls = [
-    'https://www.dropbox.com/scl/fi/34yt1vrl4mc4n9ujdf9gm/all_76.jpg?rlkey=f7b6nq478r2m9yahcalzjzif5&dl=1',
-    'https://www.dropbox.com/scl/fi/lns6cewinp7rgf3v2g1n8/all_5.jpg?rlkey=20zvut81b829k9lg5yk8ve99z&dl=1',
-    'https://www.dropbox.com/scl/fi/13jr2f1znuzulmsyabl2f/long3.jpg?rlkey=jeyriw5a8c0t42e7y2986y53m&dl=1',
-    'https://www.dropbox.com/scl/fi/nglwcza7msjo1vu4kw27r/pot4.jpg?rlkey=1ynm35b4j100ta0p5g3fx7hx4&dl=1',
-    'https://www.dropbox.com/s/7sjfwncffg8xej2/video_7.mp4?dl=1'
+    'https://www.dropbox.com/scl/fi/5pavu4vvkprrtkwktvei7/DSC02373.JPG?rlkey=fpj636qtkf3vrqfxy45n2d9ii&dl=1',
+    'https://www.dropbox.com/scl/fi/56pbn4r3ohk85rchcvwdj/DSC02813.JPG?rlkey=jnbsidqtthk6p4ysld6o6kc4t&dl=1',
+    'https://www.dropbox.com/scl/fi/av9g5zbmrrzg9064zivat/image_2.jpg?rlkey=ldocvzz5lq98zffqf1lmhbhv1&dl=1',
+    'https://www.dropbox.com/scl/fi/izo2eqqnqzcsaxis1qrbx/IMG_7612.JPG?rlkey=6wfjaux44khtlx454ex0ng0hp&dl=1',
+    'https://www.dropbox.com/scl/fi/e6vgy1et6vjr61uypk5yu/VID-20230809-WA0021.mp4?rlkey=khv8rw074vezzlg8ob38bpmbx&dl=1'
 ]
 
-# def download_file(url, save_name):
-#     url = url
-#     if not os.path.exists(save_name):
-#         file = requests.get(url)
-#         open(save_name, 'wb').write(file.content)
+def download_file(url, save_name):
+    url = url
+    if not os.path.exists(save_name):
+        file = requests.get(url)
+        open(save_name, 'wb').write(file.content)
  
-# for i, url in enumerate(file_urls):
-#     if 'mp4' in file_urls[i]:
-#         download_file(
-#             file_urls[i],
-#             f"video.mp4"
-#         )
-#     else:
-#         download_file(
-#             file_urls[i],
-#             f"image_{i}.jpg"
-#         )
+for i, url in enumerate(file_urls):
+    if 'mp4' in file_urls[i]:
+        download_file(
+            file_urls[i],
+            f"video.mp4"
+        )
+    else:
+        download_file(
+            file_urls[i],
+            f"image_{i}.jpg"
+        )
 
 
 model = YOLO('best.pt')
-# path  = [['image_0.jpg'], ['image_1.jpg'], ['image_2.jpg'], ['image_3.jpg']]
+path  = [['image_0.jpg'], ['image_1.jpg'], ['image_2.jpg'], ['image_3.jpg']]
 
-path = [['IMG_7612.JPG'], ['IMG_7678.JPG'], ['all_33.jpg'], ['all_80.jpg'], 
-    ['DSC02813.JPG'], ['DSC02373.JPG']]
+# path = [['IMG_7612.JPG'], ['IMG_7678.JPG'], ['all_33.jpg'], ['all_80.jpg'], 
+#     ['DSC02813.JPG'], ['DSC02373.JPG']]
 
 
 # path  = [['sc_1_0 (1) (1).JPG'], ['sc_1_0 (16) (1).JPG'], 
 #         ['sc_1_0 (18) (1).JPG'], ['sc_1_0 (18).JPG']]
 
-video_path = [['VID-20230809-WA0021.mp4'], ['VID-20230809-WA0022.mp4'], 
-                ['VID-20230809-WA0024.mp4'], ['VID-20230809-WA0032.mp4']]
+video_path = [['video.mp4']]
 
 classes = ['alligator_cracking', 'longitudinal_cracking', 'potholes', 'ravelling']
 
 def show_preds_image(image_path):
     image = cv2.imread(image_path)
-    outputs = model.predict(source=image_path, agnostic_nms=True, conf=0.25, iou=0.4, imgsz=1024)
+    outputs = model.predict(source=image_path, agnostic_nms=True, conf=0.25, iou=0.4, imgsz=640)
     results = outputs[0].cpu().numpy()
 
     re_boxes = results.boxes.data.tolist()
@@ -118,7 +117,7 @@ def show_preds_video(video_path):
         ret, frame = cap.read()
         if ret:
             frame_copy = frame.copy()
-            outputs = model.predict(source=frame, agnostic_nms=True, conf=0.25, iou=0.4, imgsz=1024)
+            outputs = model.predict(source=frame, agnostic_nms=True, conf=0.25, iou=0.4, imgsz=640)
             results = outputs[0].cpu().numpy()
             re_boxes = results.boxes.data.tolist()