libtorch_RefineDet_2020

libtorch_RefineDet

Requirements

• Python 3.5
• pytorch 1.0.1 py3.5_cuda8.0.61_cudnn7.1.2_2 pytorch

Introduction 训练pytorch Refinedet 环境安装步骤

• conda create -n libtorch1.0 python=3.5
• source activate libtorch1.0
• conda install pytorch torchvision cudatoolkit=8.0 -c pytorch
• pip install opencv-python

ToDo List

• trace pt

libtorch 常用api Link

[https://blog.csdn.net/yang332233/article/details/106199180]