v04 update

README.md

@@ -4,7 +4,7 @@ emoji: 🚀
 colorFrom: red
 colorTo: red
 sdk: gradio
-sdk_version: 3.0.6
+sdk_version: 3.0.9
 app_file: app.py
 pinned: false
 license: gpl-3.0

app.py

@@ -227,6 +227,12 @@ def yolo_det_img(img, device, model_name, infer_size, conf, iou, max_num, model_
     img_size = img.size  # frame size
 
     results = model(img, size=infer_size)  # detection
+    
+    # ----------------目标裁剪----------------
+    crops = results.crop(save=False)
+    img_crops = []
+    for i in range(len(crops)):
+        img_crops.append(crops[i]["im"][..., ::-1])
 
     # Data Frame
     dataframe = results.pandas().xyxy[0].round(2)
@@ -320,7 +326,7 @@ def yolo_det_img(img, device, model_name, infer_size, conf, iou, max_num, model_
     for k, v in clsDet_dict.items():
         clsRatio_dict[k] = v / clsDet_dict_sum
 
-    return det_img, objSize_dict, clsRatio_dict, det_json, report, dataframe
+    return det_img, img_crops, objSize_dict, clsRatio_dict, dataframe, det_json, report
 
 
 # YOLOv5 video detection function
@@ -517,12 +523,13 @@ def main(args):
                               label="List of detection information")
     outputs_objSize = gr.Label(label="Object size ratio statistics")
     outputs_clsSize = gr.Label(label="Category detection proportion statistics")
+    outputs_crops = gr.Gallery(label="Object crop")
 
     # -------------------output component-------------------
     outputs_video = gr.Video(format='mp4', label="Detection video")
 
     # output parameters
-    outputs_img_list = [outputs_img, outputs_objSize, outputs_clsSize, outputs_json, outputs_pdf, outputs_df]
+    outputs_img_list = [outputs_img, outputs_crops, outputs_objSize, outputs_clsSize, outputs_df, outputs_json, outputs_pdf]
     outputs_video_list = [outputs_video]
 
     # title
@@ -559,7 +566,7 @@ def main(args):
             "cpu",
             "yolov5m",
             640,
-            0.25,
+            0.6,
             0.5,
             15,
             ["person", "tie"],
@@ -592,13 +599,14 @@ def main(args):
     )
 
     gyd_video = gr.Interface(
+        # fn=yolo_det_video_test,
         fn=yolo_det_video,
         inputs=inputs_video_list,
         outputs=outputs_video_list,
         title=title,
         description=description,
         # article=article,
-        # examples=examples,
+        examples=examples,
         # theme="seafoam",
         # live=True, # Change output in real time
         flagging_dir="run",  # output directory