上传者: 42097450
|
上传时间: 2021-10-27 19:52:27
|
文件大小: 85KB
|
文件类型: -
GIthub使用指北:
1.想将项目拷贝到自己帐号下就fork一下.
2.持续关注项目更新就star一下
3.watch是设置接收邮件提醒的.
Retinanet-Pytorch
目标检测算法pytorch实现,
本项目不是完全的复现论文(很多参数以及实现方式上与原论文存在部分差异,有疑问欢迎issues)
由于一些原因,训练已经过测试,但是并没有训练完毕,所以不会上传预训练模型.
但项目代码验证无误.(但在使用时需要自己进行调整。不建议新手进行尝试。)
项目在架构上与 采用了相似的结构.
重用了大量中代码,如训练器,测试器等.
本项目单机多卡,通过torch.nn.DataParallel实现,将单机环境统一包装.支持单机单卡,单机多卡,指定gpu训练及测试,但不支持多机多卡和cpu训练和测试.
不限定检测时的设备(cpu,gpu均可).
Requirements
pytorch
op