pytorch入门项目,包括线性回归、垃圾分类、水果目标检测、ssd
人工智能导论大作业
Part2:练手小项目 线性回归和Resnet34
Part3:图片分类 基于CNN的垃圾分类 基于Resnet34的垃圾分类
Part4:目标检测 基于SSD的水果检测 其中: 数据集:fruit-detection (VOC格式) 当前训练checkpoint已删除 需重新下载vgg16预训练模型,重新训练
Step1: python creat_txt.py Step2: python creat_data_list.py #以下注意修改 checkpoint 的保存读取路径 Step3: python train.py Step4: python eval.py(所有图片评估) S tep5: python detect.py(单张图片)