THU-MIG / yolov10

YOLOv10: Real-Time End-to-End Object Detection
https://arxiv.org/abs/2405.14458
GNU Affero General Public License v3.0
8k stars 662 forks source link

The prediction is that the config.yaml configuration file is not used #303

Open KingBoyAndGirl opened 1 week ago

KingBoyAndGirl commented 1 week ago
        img_path_tmp = 'D:/code/python/object-recognition/000000003.jpg'
        local_out_file = 'D:/code/python/object-recognition/000000003.jpg_vis.jpg'
        model = YOLOv10("D:\code\python\object-recognition\src\core\mmdetection\weights\yolov10n.pt")
        predict_result = model.predict(cfg=local_yolo_config_path, source=img_path_tmp, device=device,
                                           )
        predict_result[0].save(filename=local_out_file)
        predict_result[0].show()
# Visualize settings ---------------------------------------------------------------------------------------------------
show: false # (bool) show predicted images and videos if environment allows
save_frames: false # (bool) save predicted individual video frames
save_txt: false # (bool) save results as .txt file
save_conf: false # (bool) save results with confidence scores
save_crop: false # (bool) save cropped images with results
show_labels: false # (bool) show prediction labels, i.e. 'person'
show_conf: false # (bool) show prediction confidence, i.e. '0.99'
show_boxes: false # (bool) show prediction boxes
line_width: # (int, optional) line width of the bounding boxes. Scaled to image size if None.

000000003 jpg_vis

When I configure the following configuration item as false, it does not take effect: Show_labels: false. Show_boxes: false