yolov10_CTC / yolov10 /predict.py
xiaoming32236046's picture
Upload 724 files
b9c49f5 verified
raw
history blame
369 Bytes
from ultralytics import YOLOv10
model = YOLOv10("runs/detect/train2/weights/best.pt")
results = model.predict("/ddn/imu_tsxm1/xm/GPT4V/lung/test/images/1.3.6.1.4.1.14519.5.2.1.6655.2359.100252744722628238349627337979.png")
results[0].show()
# εœ¨δ½ ηš„ YOLOv10 ζŽ¨η†δ»£η δΉ‹εŽ
result_image = results[0] # δ½ ηš„ε€„η†εŽηš„ε›Ύεƒ
result_image.save('output.png')