-
refinedet.pytorch:RefineDet的PyTorch实现下载
资源介绍
PyTorch中的RefineDet
这是的的PyTorch实施，这是张世峰，温龙吟，小编，甄磊和Stan Z.Li在CVPR2018中所做的工作。 官方的和原始的Caffe代码可以在找到。 此实现主要引用Caffe 的官方RefineDet和SSD 的PyTorch实现。 它的子库（ libs / datasets ）从进行了少量修改，即可处理数据集。 关于这个项目的中文简单帖子在。
普普派人
的Python 3.6.1
PyTorch 1.2.0
CUDA 9.0或更高
准备
首先，克隆项目并创建两个文件夹。 “数据”用于预训练的模型和数据集。 “输出”用于输出模型
git clone https://github.com/dd604/refinedet.pytorch.git
cd refinedet.pytorch
mkdir data
mkdir output
汇编
如有必