Added weights, images and videos

app.py

@@ -41,18 +41,34 @@ def letterbox(im, new_shape=(640, 640), color=(114, 114, 114), auto=True, scaleu
     im = cv2.copyMakeBorder(im, top, bottom, left, right, cv2.BORDER_CONSTANT, value=color)  # add border
     return im, r, (dw, dh)
 
-names = ['person', 'bicycle', 'car', 'motorcycle', 'airplane', 'bus', 'train', 'truck', 'boat', 'traffic light', 
-         'fire hydrant', 'stop sign', 'parking meter', 'bench', 'bird', 'cat', 'dog', 'horse', 'sheep', 'cow', 
-         'elephant', 'bear', 'zebra', 'giraffe', 'backpack', 'umbrella', 'handbag', 'tie', 'suitcase', 'frisbee', 
-         'skis', 'snowboard', 'sports ball', 'kite', 'baseball bat', 'baseball glove', 'skateboard', 'surfboard', 
-         'tennis racket', 'bottle', 'wine glass', 'cup', 'fork', 'knife', 'spoon', 'bowl', 'banana', 'apple', 
-         'sandwich', 'orange', 'broccoli', 'carrot', 'hot dog', 'pizza', 'donut', 'cake', 'chair', 'couch', 
-         'potted plant', 'bed', 'dining table', 'toilet', 'tv', 'laptop', 'mouse', 'remote', 'keyboard', 'cell phone', 
-         'microwave', 'oven', 'toaster', 'sink', 'refrigerator', 'book', 'clock', 'vase', 'scissors', 'teddy bear', 
-         'hair drier', 'toothbrush']
-
-
-colors = [[random.randint(0, 255) for _ in range(3)] for _ in names]
+names = ["animal",
+"autorickshaw",
+"bicycle",
+"bus",
+"car",
+"motorcycle",
+"person",
+"rider",
+"traffic light",
+"traffic sign",
+"truck"
+]
+
+
+#colors = [[random.randint(0, 255) for _ in range(3)] for _ in names]
+colors  = {
+    "animal": [246,198, 145],
+    "autorickshaw": [255,204, 54],
+    "bicycle": [119,11, 32],
+    "bus": [ 0,60,100],
+    "car": [ 0,0,142],
+    "motorcycle": [ 0,0,230],
+    "person": [220,20, 60],
+    "rider": [255,0, 0],
+    "traffic light": [250,170, 30],
+    "traffic sign": [220,220, 0],
+    "truck": [ 0,0, 70]
+}
 
 
 def detect(img,model,device,iou_threshold=0.45,confidence_threshold=0.25):   
@@ -134,12 +150,13 @@ def inference2(video,model_link,iou_threshold,confidence_threshold):
 
 
 
-examples_images = ['data/images/horses.jpg',
+examples_images = ['data/images/1.jpg',
+            'data/images/2.jpg',
             'data/images/bus.jpg',
-            'data/images/zidane.jpg']
-examples_videos = ['data/video/input_0.mp4','data/video/input_1.mp4'] 
+            'data/images/3.jpg']
+examples_videos = ['data/video/1.mp4','data/video/2.mp4'] 
 
-models = ['yolov7','yolov7x','yolov7-w6','yolov7-d6','yolov7-e6e']
+models = ['yolov7','yolov7t']
 
 with gr.Blocks() as demo:
     gr.Markdown("## YOLOv7 Inference")
@@ -157,8 +174,8 @@ with gr.Blocks() as demo:
     with gr.Tab("Video"):
         gr.Markdown("## YOLOv7 Inference on Video")
         with gr.Row():
-            video_input = gr.Video(type='pil', label="Input Image", source="upload")
-            video_output = gr.Video(type="pil", label="Output Image",format="mp4")
+            video_input = gr.Video(type='pil', label="Input Video", source="upload")
+            video_output = gr.Video(type="pil", label="Output Video",format="mp4")
         fps_video = gr.Number(0,label='FPS')
         video_drop = gr.Dropdown(choices=models,value=models[0])
         video_iou_threshold = gr.Slider(label="IOU Threshold",interactive=True, minimum=0.0, maximum=1.0, value=0.45)
@@ -166,9 +183,9 @@ with gr.Blocks() as demo:
         gr.Examples(examples=examples_videos,inputs=video_input,outputs=video_output)
         video_button = gr.Button("Detect")
     
-    with gr.Tab("Webcam Video"):
-        gr.Markdown("## YOLOv7 Inference on Webcam Video")
-        gr.Markdown("Coming Soon")
+    # with gr.Tab("Webcam Video"):
+    #     gr.Markdown("## YOLOv7 Inference on Webcam Video")
+    #     gr.Markdown("Coming Soon")
 
     text_button.click(inference, inputs=[image_input,image_drop,
                                          image_iou_threshold,image_conf_threshold],

data/video/{input_0.mp4 → 1.mp4}

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:0305b8ba574b0b26c8d21b70dbd80aacbba13e813a99f5c4675982063ebd18a3
-size 1103987
+oid sha256:271041b71b046d282746f21bd857b2abc33302fed1f0818a7066880db6eff9a3
+size 14396787

data/video/{input_1.mp4 → 2.mp4}

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:103c54e60a7abf381741b4bcc5e696f5d200fc2a4ff259cc779bbd511dc2dae1
-size 1792081
+oid sha256:4d755c21948e68245f43709b1219a1355a73e7644329c6527e7430ad7f53dffe
+size 10771315

weights/yolov7-e6e.pt

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:b370120a414bf32b5d65fc808e5a32c8d9b3c63902d1bc41894fc9d86eccf9cb
-size 304425133

weights/yolov7-w6.pt

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:2be96abd35939335e5eb08bcde2685ae44c3fadd8af7d601734ba10c8737425c
-size 141255765

weights/yolov7.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d1ac3c74eb96a3eec77949c0f37a06bc272756606ff011d9353f7abff4e0c71d
-size 75587165
+oid sha256:0f710202260d40caed72d1ca0e4941d13b085a9421dc6af921de934a59be8dc4
+size 298803327

weights/{yolov7-d6.pt → yolov7t.pt}

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:154bb05fceb4f06f062a085a7f3d99d006cf5f1f53237c3a1bdbd66b0ba1a74c
-size 268261125
+oid sha256:79cc87831d4ec0c9a95a1c657da02480af47b0e0b268e98245ba72093b1cbc27
+size 12328037

weights/yolov7x.pt

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:4873202bc3fa8b1d4c4865a37b7c61ce7b7b1cdbe6f31017c33be2bd11826e98
-size 143099649