remove useless codes

Author: JudasDie

experiments/train/Ocean.yaml

@@ -1,6 +1,5 @@
 OCEAN:
-#  GPUS: '0,1,2,3,4,5,6,7'
-  GPUS: '0'
+  GPUS: '0,1,2,3,4,5,6,7'
   PRINT_FREQ: 10
   WORKERS: 32
   OUTPUT_DIR: 'logs'   # log file

lib/models/models.py

@@ -7,8 +7,8 @@
 import torch
 import torch.nn as nn
 from .ocean import Ocean_
-from .oceanplus import OceanPlus_
-from .oceanplusTRT import OceanPlusTRT_
+#from .oceanplus import OceanPlus_
+#from .oceanplusTRT import OceanPlusTRT_
 from .oceanTRT import OceanTRT_
 from .siamfc import SiamFC_
 from .connect import box_tower, AdjustLayer, AlignHead, Corr_Up, MultiDiCorr, OceanCorr
@@ -49,24 +49,24 @@ def __init__(self, online=False, align=False):
         self.connect_model2 = OceanCorr()
 
 
-class OceanPlus(OceanPlus_):
-    def __init__(self, online=False):
-        super(OceanPlus, self).__init__()
-        self.features = ResNet50(used_layers=[3], online=online)   # in param
-        self.neck = AdjustLayer(in_channels=1024, out_channels=256)
-        self.connect_model = box_tower(inchannels=256, outchannels=256, towernum=4)
-        self.mask_model = MultiRefine(addCorr=True, mulOradd='add')
+#class OceanPlus(OceanPlus_):
+#    def __init__(self, online=False):
+#        super(OceanPlus, self).__init__()
+#        self.features = ResNet50(used_layers=[3], online=online)   # in param
+#        self.neck = AdjustLayer(in_channels=1024, out_channels=256)
+#        self.connect_model = box_tower(inchannels=256, outchannels=256, towernum=4)
+#        self.mask_model = MultiRefine(addCorr=True, mulOradd='add')
 
 
-class OceanPlusTRT(OceanPlusTRT_):
-    def __init__(self, online=False):
-        super(OceanPlusTRT, self).__init__()
-        self.features = ResNet50(used_layers=[3], online=online)  # in param
-        self.neck = AdjustLayer(in_channels=1024, out_channels=256)
-        self.connect_model0 = MultiDiCorr(inchannels=256, outchannels=256)
-        self.connect_model1 = box_tower(inchannels=256, outchannels=256, towernum=4)
-        self.connect_model2 = OceanCorr()
-        self.mask_model = MultiRefineTRT(addCorr=True, mulOradd='add')
+#class OceanPlusTRT(OceanPlusTRT_):
+#    def __init__(self, online=False):
+#        super(OceanPlusTRT, self).__init__()
+#        self.features = ResNet50(used_layers=[3], online=online)  # in param
+#        self.neck = AdjustLayer(in_channels=1024, out_channels=256)
+#        self.connect_model0 = MultiDiCorr(inchannels=256, outchannels=256)
+#        self.connect_model1 = box_tower(inchannels=256, outchannels=256, towernum=4)
+#        self.connect_model2 = OceanCorr()
+#        self.mask_model = MultiRefineTRT(addCorr=True, mulOradd='add')
 
 
 # ------------------------------
@@ -348,4 +348,4 @@ def ONLINEnet50(filter_size=4, optim_iter=5, optim_init_step=0.9, optim_init_reg
 
     # ONLINE network
     net = ONLINEnet(feature_extractor=backbone_net, classifier=classifier, classification_layer=classification_layer)
-    return net
+    return net

lib/utils/utils.py

@@ -1145,7 +1145,7 @@ def _build_lr_scheduler(optimizer, config, epochs=50, last_epoch=-1):
 
 
 def _build_warm_up_scheduler(optimizer, cfg, epochs=50, last_epoch=-1, modelFLAG='OCEAN'):
-    cfg = cfg[modelFLAG]
+    #cfg = cfg[modelFLAG]
     warmup_epoch = cfg.TRAIN.WARMUP.EPOCH
     sc1 = _build_lr_scheduler(optimizer, cfg.TRAIN.WARMUP,
                               warmup_epoch, last_epoch)

tracking/train_ocean.py

@@ -128,8 +128,7 @@ def build_opt_lr(cfg, model, current_epoch=0):
         trainable_params += [{'params': model.align_head.parameters(),
                             'lr': cfg.OCEAN.TRAIN.BASE_LR}]
     except:
-        trainable_params += [{'params': model.align_head.parameters(),
-                              'lr': cfg.OCEAN.TRAIN.BASE_LR}]
+        pass
 
     # print trainable parameter (first check)
     print('==========first check trainable==========')