ssds.pytorch:单发MultiBox检测器及其变体的存储库,通过pytorch,python3实现

时间:2024-03-23 09:07:26
【文件属性】:

文件名称:ssds.pytorch:单发MultiBox检测器及其变体的存储库,通过pytorch,python3实现

文件大小:1.34MB

文件格式:ZIP

更新时间:2024-03-23 09:07:26

pytorch vgg yolo ssd darknet

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>


网友评论