11 lines
196 B
Python
11 lines
196 B
Python
from ultralytics import YOLO
|
|
from matplotlib import pyplot as plt
|
|
|
|
m = YOLO("best.pt")
|
|
|
|
res = m("../dataset/images/train/1-1-0000001-1.jpg")
|
|
res[0].show()
|
|
#print(res)
|
|
#plt.imshow(res)
|
|
#plt.show()
|