SSD在pytorch下尝试

一、创建环境

conda create -n pytorch python=3.7

安装pytorch

conda install pytorch torchvision cudatoolkit=9.2 -c pytorch

太慢了，换

pip install torch torchvision

激活环境

二 、部署ssd.pytorch

项目地址：https://github.com/amdegroot/ssd.pytorch

git clone https://github.com/amdegroot/ssd.pytorch.git

 下载项目

然后按照提示下载数据集，没看懂这个三个数据集的部署形式，没办法，挂着下了一宿

.txt文件是手动移过去的

下载后的数据集存放如图

但是发现数据集在主目录上，还是没法使用，建立了个连接直接跳转

ln -s /home/king/data/coco