mirrors
/
fast-reid
mirror of https://github.com/JDAI-CV/fast-reid.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

update pairwise circle loss 

Summary: add param of pairwise circle loss to config, and update pairwise circle loss version
pull/150/head
liaoxingyu 2020-06-10 19:07:29 +08:00
parent 96b9ad2d99
commit cbdc01a1c3
2 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
fastreid/config/defaults.py
View File

@ -97,6 +97,11 @@ _C.MODEL.LOSSES.TRI.HARD_MINING = True
_C.MODEL.LOSSES.TRI.USE_COSINE_DIST = False
_C.MODEL.LOSSES.TRI.SCALE = 1.0
# Circle Loss options
_C.MODEL.LOSSES.CIRCLE = CN()
_C.MODEL.LOSSES.CIRCLE.MARGIN = 0.25
_C.MODEL.LOSSES.CIRCLE.SCALE = 128
# Focal Loss options
_C.MODEL.LOSSES.FL = CN()
_C.MODEL.LOSSES.FL.ALPHA = 0.25

6
fastreid/modeling/losses/metric_loss.py
View File

@ -164,10 +164,10 @@ class TripletLoss(object):
class CircleLoss(object):
def __init__(self, cfg):
self._scale = cfg.MODEL.LOSSES.SCALE_TRI
self._scale = cfg.MODEL.LOSSES.CIRCLE.SCALE
self.m = 0.25
self.s = 128
self.m = cfg.MODEL.LOSSES.CIRCLE.MARGIN
self.s = cfg.MODEL.LOSSES.CIRCLE.SCALE
def __call__(self, _, global_features, targets):
global_features = normalize(global_features, axis=-1)