move utils_iou_eval.py to modules dir

Author: gemfield

config.py

@@ -84,7 +84,7 @@
 #config.core.jit_model_path = "<your-script-or-quantize-model-path>"
 
 ## -------------------- training ------------------
-config.core.epoch_num = 200
+config.core.epoch_num = 100
 config.core.save_num = 1
 config.core.shuffle = True
 config.core.batch_size = 16
@@ -105,20 +105,12 @@
 weight = torch.from_numpy(config.data['classWeights']).to(config.core.device)
 config.core.criterion = torch.nn.CrossEntropyLoss(weight)
 
-# config.core.teacher = AttrDict()
-# config.core.teacher.net = EESPNet_Seg(config.core.cls_num)
-# config.core.teacher.criterion = torch.nn.CrossEntropyLoss(weight)
-# config.core.teacher.optimizer = torch.optim.Adam(config.core.teacher.net.parameters(), 5e-4, (0.9, 0.999), eps=1e-08, weight_decay=5e-4)
-
 config.core.mean = config.data['mean']
 config.core.std = config.data['std']
 ## -------------------- optimizer and scheduler ------------------
 config.core.optimizer = torch.optim.Adam(config.core.net.parameters(), 3e-4, (0.9, 0.999), eps=1e-08, weight_decay=5e-4)
-# config.core.optimizer = torch.optim.SGD(config.core.net.parameters(), 5e-3, momentum=0.9)
-
 lambda_lr = lambda epoch: round ((1 - epoch/config.core.epoch_num) ** 0.9, 8)
 config.core.scheduler = optim.lr_scheduler.LambdaLR(config.core.optimizer, lr_lambda=lambda_lr)
-# config.core.scheduler = optim.lr_scheduler.MultiStepLR(config.core.optimizer, milestones=[20,40,55,70,80,90,100,110,120,130,140,150,160], gamma=0.27030)
 
 ## -------------------- loader ------------------
 config.core.num_workers = 3
@@ -182,4 +174,4 @@
     batch_size=config.core.batch_size, shuffle=True, num_workers=config.core.num_workers, pin_memory=config.pin_memory)
 scale4_train_loader.is_last_loader = False
 # multi loader
-config.core.train_loader_list = [scale1_train_loader, scale2_train_loader, scale4_train_loader, scale3_train_loader, last_train_loader]
+config.core.train_loader_list = [scale1_train_loader, scale2_train_loader, scale4_train_loader, scale3_train_loader, last_train_loader]

utils/utils_IOU_eval.py renamed to modules/utils_IOU_eval.py

@@ -8,7 +8,7 @@
 import torch.nn.functional as F
 import deepvac
 from deepvac import LOG, Deepvac
-from utils.utils_IOU_eval import IOUEval
+from modules.utils_IOU_eval import IOUEval
 from data.dataloader import OsWalkDataset2
 
 class ESPNetTest(Deepvac):

test.py

@@ -8,7 +8,7 @@
 from torch import nn
 from torch import optim
 from deepvac import LOG, DeepvacTrain
-from utils.utils_IOU_eval import IOUEval
+from modules.utils_IOU_eval import IOUEval
 
 class ESPNetTrain(DeepvacTrain):
     def __init__(self, deepvac_config):

train.py

@@ -10,7 +10,7 @@
 import deepvac
 from deepvac import LOG, DeepvacTrain
 from deepvac.experimental.core import DeepvacDistill
-from utils.utils_IOU_eval import IOUEval
+from modules.utils_IOU_eval import IOUEval
 
 class ESPNetTrain(DeepvacDistill):
     def __init__(self, deepvac_config):