栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

Ubuntu16.04配置deeplabv3+的pytorch版本

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

Ubuntu16.04配置deeplabv3+的pytorch版本

deeplabv3+的github代码:https://github.com/VainF/DeepLabV3Plus-Pytorch
1.我的环境配置
conda安装cud11.0+高版本8.0的cudnn

scipy==1.7.1
numpy==1.21.2
matplotlib==3.4.3
opencv-python==4.5.3.56
torch==1.7.1
torchvision==0.8.2
tqdm==4.62.2
Pillow==8.2.0
h5py==2.10.0
visdom

2.预训练权重下载
腾讯微云:https://share.weiyun.com/qqx78Pv5
讲文件放在新建的文件夹checkpoints下
3.预测图片
首先进入新建的环境

conda activate py1.7

预测结果保存在新建文件夹test_results下
单张图片预测

python predict.py --input 1.png  --dataset voc --model deeplabv3plus_mobilenet --ckpt checkpoints/best_deeplabv3plus_mobilenet_voc_os16.pth --save_val_results_to test_results

文件夹下所有图片预测

python predict.py --input samples  --dataset voc --model deeplabv3plus_mobilenet --ckpt checkpoints/best_deeplabv3plus_mobilenet_voc_os16.pth --save_val_results_to test_results


参考:https://blog.csdn.net/weixin_46118817/article/details/125121751

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/965885.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号