| 1234567891011121314151617181920212223 |
- import warnings
- warnings.filterwarnings('ignore')
- from ultralytics import YOLO
- # 推理参数官方详解链接：https://docs.ultralytics.com/modes/predict/#inference-sources:~:text=of%20Results%20objects-,Inference%20Arguments,-model.predict()
- if __name__ == '__main__':
- model = YOLO('runs/train/exp/weights/best.pt') # select your model.pt path
- model.predict(source='dataset/images/test',
- imgsz=640,
- project='runs/detect',
- name='exp',
- save=True,
- # conf=0.2,
- # iou=0.7,
- # agnostic_nms=True,
- # visualize=True, # visualize model features maps
- # line_width=2, # line width of the bounding boxes
- # show_conf=False, # do not show prediction confidence
- # show_labels=False, # do not show prediction labels
- # save_txt=True, # save results as .txt file
- # save_crop=True, # save cropped images with results
- )
|
