YOLO model detection as a regression problem [17]. Thus the input image(06) Save Image Image gallery: Yolo Self.model.predict To Cpu