3fd025a
1
2
3
4
5
6
from ultralytics import YOLO model_init= YOLO('model/pepsodent.pt').cuda() model = model_init.to(device=0)