
repulsion_loss_ssd: 排斥损失
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
Repulsion_Loss_SSD提出了一种新颖的排斥损失函数,旨在优化单阶段目标检测模型SSD的性能,通过增强类别间的区分度来提高小目标及拥挤场景下的检测精度。
使用SSD实施排斥力损失的实现基于Wei Wei, Dragomir Anguelov, Dumitru Erhan, Christian Szegedy, Scott Reed, Cheng-Yang及Alexander C.Berg于2016年发表的论文。官方原始Caffe代码可在相关页面获取。
安装目录:选择适合您的环境并运行适当的命令以在网站上安装PyTorch 。克隆此存储库后,请按照说明下载数据集。现在,我们支持Visdom用于训练过程中的实时损失可视化!要在浏览器中使用Visdom,请执行以下操作:
# 首先安装Python服务器和客户端
pip install visdom
# 启动服务端(建议在screen或tmux会话中运行)
全部评论 (0)
还没有任何评论哟~


