ajaxmulticommentary / layout-yolo-models Goto Github PK

import torch
model = torch.hub.load('.', 'custom', path='/path/to/yolov5/runs/train/exp5/weights/best.pt', source='local') 
# Image
img = '/path/to/test/image/25.jpg'
# Inference
results = model(img)
# Results, change the flowing to: results.show()
results.show()  # or .show(), .save(), .crop(), .pandas(), etc