我是靠谱客的博主 外向战斗机，最近开发中收集的这篇文章主要介绍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 18.04使用pytorch跑SSD+Mobilenet V2安装pytorch安装visdom下载代码下载数据集下载预训练模型修改显卡索引所遇到的程序开发问题。
如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。
本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。
发表评论 取消回复