校内实习项目 SSD-PyTorch GUI
- 项目文件ssd.pytorch-master-withgui.rar解压
- 解压数据集到以下路径 文件夹命名为\VOCdevkit
包括两组 VOC2007 和 VOC2012
另外需要修改项目文件中ssd.pytorch-master/voc0712.py中的VOC_ROOT为数据集存放路径
只采用VOC2007数据集也可以进行训练,只需修改ssd.pytorch-master/voc0712.py中第102行代码为
image_sets=[('2007', 'trainval')],
- 配置Anaconda + PyCharm 环境
在Anaconda安装路径下的 \envs
解压虚拟环境压缩包tesr.rar (相当于创建虚拟环境,但是不必在线下载包)
可在Anaconda Prompt 中使用命令查看是否存在虚拟环境test
conda env list
- 为项目配置运行环境
查看当前python文件是否运行在所需的环境中
不是的话可以进行选择
- 修改项目中的权重路径
改为项目文件夹下的weights文件夹
- 运行window.py