上传者: 42106765
|
上传时间: 2021-12-08 14:50:27
|
文件大小: 1.34MB
|
文件类型: -
ssds.pytorch
适用于pytorch,python3的Single Shot MultiBox Detector及其变体的存储库。 此存储库易于设置,并具有大量可视化方法。 我们希望此回购协议可以帮助人们更好地理解类似ssd的模型,并帮助人们轻松地训练和部署ssds模型。
当前,它包含以下功能:
多种SSD变体:ssd,fpn,bifpn,yolo等
多基础网络:resnet,regnet,mobilenet等。
可视化类似于ssd的模型的功能,以帮助用户了解模型的设计和性能。
快速培训和推论:利用Nvidia Apex和Dali进行快速培训,并支持用户将模型转换为ONNX或TensorRT进行部署。
此回购取决于 , 和。 感谢他们的工作。
注意当前版本的预训练模型尚未完成,请检查以获取丰富的预训练模型。
目录
安装
要求
python> = 3.7
CUDA>