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()