update

liuning-scu-cn · Aug 12, 2019 · 449d90c · 449d90c
1 parent 08206b0
commit 449d90c
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/core/model.py b/core/model.py
@@ -868,7 +868,9 @@ def forward(self, x, label=None, direction=None):
         backbone_f = x.detach()
         x = self.fc7(x)
 
+        # TigerID
         glogit = self.cls(x)
+        # Left/Right
         dlogit = self.cls_direction(x)
 
         # erase
@@ -883,15 +885,19 @@ def forward(self, x, label=None, direction=None):
         erase_f = x1.detach()
         x1 = self.erase_fc7(x1)
 
+        # TigerID
         erase_glogit = self.erase_cls(x1)
+        # Left/Right
         erase_dlogit = self.erase_cls_direction(x1)
 
         # fuse
         fuse_f = torch.cat((backbone_f, erase_f), dim=1)
         x2 = self.fuse_fc7(fuse_f)
         x2 = l2_norm(x2)
 
+        # TigerID
         fuse_glogit = self.fuse_cls(x2)
+        # Left/Right
         fuse_dlogit = self.fuse_cls_direction(x2)
 
         return [glogit, dlogit, erase_glogit, erase_dlogit, fuse_glogit, fuse_dlogit, x2]