如何再Python專案中調用訓練好的yolov5模型

yolov5 python 影像辨識 python實作

andyvcandyburst 2023-01-11 00:04:40 ‧ 1471 瀏覽

想請問如何在其他Python專案中調用已經訓練好的YOLO模型
需要先對訓練好的模型(best.pt)做甚麼動作嗎另存之類的然後在其他地方才可以調用因為第一次做這種東西還不太懂想請各位大大幫忙~~

直播研討會

{{ item.channelVendor }} {{ item.webinarstarted }} |

直播中

hokou

iT邦好手 1 級 ‧ 2023-01-11 08:20:19

最佳解答

看來是 PyTorch 的 Model
如果不知道 PyTorch 可能要先花點時間去理解了

# Export:

model_scripted = torch.jit.script(model) # Export to TorchScript
model_scripted.save('model_scripted.pt') # Save

# Load:

model = torch.jit.load('model_scripted.pt')
model.eval()

立即登入回答

如何再Python專案中 調用訓練好的yolov5模型