1 Star 0 Fork 0

loic_wang/ZJUAI_PyramidBoxDetector

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
该仓库未声明开源许可证文件(LICENSE),使用请关注具体项目描述及其代码上游依赖。
克隆/下载
README 1.52 KB
一键复制 编辑 原始数据 按行查看 历史
叶伟才 提交于 2020-05-28 15:37 . Update README
2018云从人头技术 冠军方案分享
运行环境:
Ubuntu16.04
pytorch0.3.1 torchvision numpy
cuda cudnn等
需要GPU 如Titan 12G等。
pretrained_model/Res50_pyramid.pth
链接:https://pan.baidu.com/s/13Uf1uWSi-aMbNsbzUDl5Gg 密码:yy73
Test
新建weights/, model文件在这里下载
通用模型: 链接:https://pan.baidu.com/s/1hDGi02UsHplftYFpfcAvpw 提取码:9xk1
our专用模型:链接:https://pan.baidu.com/s/1PN3aBBr4Frzb9RvO4NgfNw 提取码:kkrc
1、将数据置于yuncong_test_set文件夹中,
2、直接运行test_yuncong_*.py 可以得到对应的生成文件,更多的参数设置请看文件
train
1 将数据放在当前目录下,即2018云从人头技术-...., 此目录下有Mall Part_A 等场景的原始数据,你可以相应的调整,但原来的代码文件路径可能需要相应的调整.
2 数据增强,对每个场景都有对应的数据增强,由于之前使用的是paddlepaddle版本的pyramidbox的数据格式,需要调用一下topytorch.py来生成pytorch版本的数据格式,可以自己去调整. 最后对每个场景生成训练的数据文件集如: cat UCSD.txt augfile_UCSD > final_UCSD.txt
3 然后可以进行训练, 可生成通用的模型,也可对应场景的模型, mod/parta/partb/mall/ucsd/our等. 运行相应的代码即可.
最近在赶deadline,更多训练的步骤,后续将更详细的给出.
如有疑问,请联系我
微信: 13032893650
手机: 13032893650
email: 2755289083@qq.com yeweicai@zju.edu.cn
更多细节请看 .docx 和 .pptx 文件
欢迎star
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/loic_wang/ZJUAI_PyramidBoxDetector.git
git@gitee.com:loic_wang/ZJUAI_PyramidBoxDetector.git
loic_wang
ZJUAI_PyramidBoxDetector
ZJUAI_PyramidBoxDetector
master

搜索帮助