上传者: 42097533
|
上传时间: 2022-03-04 14:38:43
|
文件大小: 2.89MB
|
文件类型: -
从带代码进行一些自定义更改
ssds.pytorch
适用于pytorch,python3的Single Shot MultiBox Detector及其变种的存储库。
当前,它包含以下功能:
多种SSD变体:ssd,rfb,fssd,ssd-lite,rfb-lite,fssd-lite
多基础网络:VGG,Mobilenet V1 / V2
免费图片大小
使用可视化:训练损失,评估损失/ mAP,例如弓箭手箱。
此存储库取决于 , , , 和。 谢谢你的工作。
目录
安装
安装pytorch
通过pip install -r ./requirements.txt安装要求pip install -r ./requirements.txt
用法
训练,测试和演示一些特定模型。 请使用模型配置文件运行文件夹中的相关文件,例如:
python train.py --cfg