pytorch 使用.pt文件分析图像

lymgl2op  于 2022-11-09  发布在  其他
关注(0)|答案(1)|浏览(195)

我真的是人工智能新手,所以很抱歉我会犯一些概念上的错误。我有一个.pt文件,我想用它来获取图片的一些边界框。最简单的方法是什么?类似于:

import torch
model=torch.load('best.pt')
img=['foto.jpg']
results = model(img)
results.show()

非常感谢!

相关问题