
pytorch
北方的杨先生
声明:博文的编写,主要参考网上资料,并结合个人见解,仅供学习、交流使用,如有侵权,请联系博主删除,原创文章转载请注明出处。博主微信:wx_big。
展开
-
smoothloss
ImageNet 上标签平滑效果的可视化。顶部:当增加ε时,目标类别与其它类别之间的理论差距减小。下图:最大预测与其它类别平均值之间差距的经验分布。很明显,通过标签平滑,分布中心处于理论值并具有较少的极端值。Loss:基于CrossEntropy进行Label Smooth操作class CrossEntropyLabelSmooth(nn.Module):ㅤdef __init__(self, num_classes, epsilon=0.1, use_gpu=True):...原创 2020-05-21 10:39:19 · 1218 阅读 · 0 评论 -
目标检测iou loss
GIOU Loss解决 iou loss 为1 时不优化的缺陷 增加c项,表示为预测框与标签框最小矩形 缺陷,当两个框相交时,尤其一个框包含另一个的时候,退化为iou loss,收敛会变慢DIOU Loss直接最小化Anchor和目标框之间的归一化距离以达到更快的收敛速度 b、bgt分别是anchor、groundtrueth中心点坐标,p为其欧式距离,c为其包含最小矩形对角线长度 解决GIOU Loss 退化IOU loss的情况CIOU Loss回归损失在与目标框有.原创 2020-05-21 10:33:37 · 1292 阅读 · 0 评论 -
torch.backends.cudnn.benchmark = true
torch.backends.cudnn.benchmark = true一般来讲,应该遵循以下准则:如果网络的输入数据维度或类型上变化不大,设置torch.backends.cudnn.benchmark=true 可以增加运行效率; 如果网络的输入数据在每次 iteration 都变化的话,会导致 cnDNN 每次都会去寻找一遍最优配置,这样反而会降低运行效率。...原创 2020-05-21 10:30:27 · 819 阅读 · 0 评论 -
人脸关键点检测SOTA模型
模型大小6M inference cpu 40ms wflw 98 个点 NE3.5,比第二名高0.9个点原创 2020-04-01 16:54:30 · 1573 阅读 · 1 评论 -
关键点评估 ION、IPN、NME
原创 2020-04-01 16:31:02 · 4959 阅读 · 2 评论 -
Pytorch 模型时间变长问题
原创 2020-04-01 16:28:24 · 1162 阅读 · 0 评论 -
pytorch tricks
您的鼓励是我前进的源动力????原创 2020-01-10 15:56:19 · 261 阅读 · 0 评论 -
Pytorch转onnx
原创 2019-08-13 15:38:25 · 2055 阅读 · 0 评论