- torch1.0
- torchvision0.2.0
- python3.5
- ubuntu16.04
- pycharm2018
- ResNet
- MobileNet
- YOLOV3
- FCN
- mnist
- asl,A-E共五类,共15000个数据,12000的训练集、3000的验证集
- VOC2012,主要用来做语义分割实验
- 训练模型:
python asl.py -v V2 -e 20
python asl.py -v V3 -t large -e 20
- demo演示:
python demo.py -v V2
- 获取yolov3的权重文件
wget https://github.com/ayooshkathuria/pytorch-yolo-v3/raw/master/dog-cycle-car.png
- 将待测试图片放入test文件夹,输出结果保存在detect文件夹中
- 执行
python demo.py
- 需要提前下载好VOC2012数据集,并按程序中的规定的目录地址放在data文件夹中
wget http://host.robots.ox.ac.uk/pascal/VOC/voc2012/VOCtrainval_11-May-2012.tar