feat(ml): added semantic threshold option

models/cycle_gan_semantic_mask_model.py

@@ -412,7 +412,7 @@ def backward_G(self):
 
         # only use semantic loss when classifier has reasonably low loss
         #if True:
-        if not hasattr(self, 'loss_f_s') or self.loss_f_s.detach().item() > 1.0:
+        if not hasattr(self, 'loss_f_s') or self.loss_f_s.detach().item() > self.opt.semantic_threshold:
             self.loss_sem_AB = 0 * self.loss_sem_AB 
             self.loss_sem_BA = 0 * self.loss_sem_BA 
         self.loss_G += self.loss_sem_BA + self.loss_sem_AB

models/cycle_gan_semantic_model.py

@@ -273,7 +273,7 @@ def backward_G(self):
 
         # only use semantic loss when classifier has reasonably low loss
         #if True:
-        if not hasattr(self, 'loss_CLS') or self.loss_CLS.detach().item() > 1.0:
+        if not hasattr(self, 'loss_CLS') or self.loss_CLS.detach().item() > self.opt.semantic_threshold:
             self.loss_sem_AB = 0 * self.loss_sem_AB 
             self.loss_sem_BA = 0 * self.loss_sem_BA 
 

options/base_options.py

@@ -64,6 +64,7 @@ def initialize(self, parser):
         parser.add_argument('--verbose', action='store_true', help='if specified, print more debugging information')
         parser.add_argument('--suffix', default='', type=str, help='customized suffix: opt.name = opt.name + suffix: e.g., {model}_{netG}_size{load_size}')
         parser.add_argument('--semantic_nclasses',default=10,type=int,help='number of classes of the semantic loss classifier')
+        parser.add_argument('--semantic_threshold',default=1.0,type=float,help='threshold of the semantic classifier loss below with semantic loss is applied')
         parser.add_argument('--display_networks', action='store_true',help='Set True if you want to display networks on port 8000')
         parser.add_argument('--compute_fid', action='store_true')
         parser.add_argument('--fid_every', type=int, default=1000)