26 lines
1.2 KiB
Python
26 lines
1.2 KiB
Python
import warnings
|
||
warnings.filterwarnings('ignore')
|
||
from ultralytics import YOLO
|
||
|
||
# BILIBILI UP 魔傀面具
|
||
# 推理参数官方详解链接：https://docs.ultralytics.com/modes/predict/#inference-sources:~:text=of%20Results%20objects-,Inference%20Arguments,-model.predict()
|
||
|
||
# 预测框粗细和颜色修改问题可看<使用说明.md>下方的<YOLOV8源码常见疑问解答小课堂>第六点
|
||
|
||
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
|
||
) |