深度学习框架中的Ranking Loss
深度学习框架中的Ranking LossCaffeConstrastive Loss Layer. 限于 Pairwise Ranking Loss 计算. 例如,可以用于训练 Siamese 网络。PyCaffe Triplet Ranking Loss Layer. 用来训练 triplet 网络,by David Lu。PyTorchCosineEmbeddingLoss. 使用余弦相似度的
·
深度学习框架中的Ranking Loss
Caffe
Constrastive Loss Layer. 限于 Pairwise Ranking Loss 计算. 例如,可以用于训练 Siamese 网络。
PyCaffe Triplet Ranking Loss Layer. 用来训练 triplet 网络,by David Lu。
PyTorch
CosineEmbeddingLoss. 使用余弦相似度的 Pairwise Loss。输入是一对二元组,标签标记它是一个正样本对还是负样本对,以及边距 margin。
MarginRankingLoss. 同上, 但使用欧拉距离。
TripletMarginLoss. 使用欧拉距离的 Triplet Loss。
TensorFlow
contrastive_loss. Pairwise Ranking Loss.
triplet_semihard_loss. 使用 semi-hard 负采样的 Triplet loss。
更多推荐
已为社区贡献1条内容
所有评论(0)