17 lines
485 B
Python
17 lines
485 B
Python
# 导入ultralytics的YOLO库
|
|
from ultralytics import YOLO
|
|
|
|
# 加载模型
|
|
model = YOLO(r'D:\ultralytics-main\ultralytics-main\runs\detect\train5\weights\best.pt') # 加载自定义的训练模型
|
|
|
|
if __name__ == '__main__':
|
|
|
|
results = model.predict(
|
|
r"D:\ultralytics-main\ultralytics-main\ultralytics\datasets\original-license-plates\test\images\b32671b7cc279583_jpg.rf.f898026e2169a3b6adc4d21aee6d06c5.jpg",
|
|
save=True, imgsz=320, conf=0.5,device=0,iou=0.5)
|
|
|
|
|
|
|
|
|
|
|