为了进一步使用c++调用deeplabv3+模型,使用trace将pytorch训练生成的.pth格式转为.pt
参考:https://github.com/shanson123/ORB_SLAM2_DeeplabV3/blob/master/DeeplabV3/create_deeplabv3.py
在predict.py文件中添加:
with torch.no_grad():
model = model.eval()
for img_path in tqdm(image_files)

最低0.47元/天 解锁文章
9282

被折叠的 条评论
为什么被折叠?



