【文件属性】:
文件名称:pytorch-learning-notes:pytorch学习教程
文件大小:615.93MB
文件格式:ZIP
更新时间:2021-05-12 10:45:37
附件源码 文章源码
pytorch-learning-tutorials
觉得可以,麻烦给个”Star“
1.image_classification 图像分类
《pytorch实现L2和L1正则化的方法》
2.object_detection目标检测
mobileNet v1 v2 SSD目标检测:该项目是参考《pytorch-ssd》 ,修改,主要是方便训练。
数据集VOC2007和VOC2012:
训练方法:my_train_ssd.py
修改my_train_ssd.py的参数train_filename和val_filename即可,直接运行训练,例如:
train_filename = 'E:/git/VOC0712_dataset/train.txt' #训练文件
val_filename = 'E:/git/VOC0712_dataset/val.txt' #测试文件
测试方法:run