我是靠谱客的博主 外向战斗机，这篇文章主要介绍Ubuntu 18.04使用pytorch跑SSD+Mobilenet V2安装pytorch安装visdom下载代码下载数据集下载预训练模型修改显卡索引，现在分享给大家，希望可以做个参考。
安装pytorch
pip3 install torch
pip3 install --no-deps torchvision
安装visdom
pip3 install visdom
python3 -m visdom.server
下载代码
git clone https://github.com/ZhongyuanW/SSD_mobilenetv2-with-Focal-loss
下载数据集
cd SSD_mobilenetv2-with-Focal-loss/data/scripts
./VOC2007.sh ~ （将数据集下载到用户根目录，如果下载慢，可以用迅雷）
./VOC2012.sh ~
下载预训练模型
1.百度网盘: https://pan.baidu.com/s/1RmOPF4jQYpYlE_8E4DifeQ 提取码: f53n
2.下载mobilenetv2.pth.tar，放到SSD_mobilenetv2-with-Focal-lossweights下
修改显卡索引
train.py中修改为os.environ["CUDA_VISIBLE_DEVICES"] = "0"
最后
以上就是外向战斗机最近收集整理的关于Ubuntu 18.04使用pytorch跑SSD+Mobilenet V2安装pytorch安装visdom下载代码下载数据集下载预训练模型修改显卡索引的全部内容，更多相关Ubuntu内容请搜索靠谱客的其他文章。
本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。
发表评论 取消回复