|
|
|
@@ -20,43 +20,6 @@ class Exp(BaseExp): |
|
|
|
self.depth = 1.00 |
|
|
|
self.width = 1.00 |
|
|
|
|
|
|
|
# ---------------- dataloader config ---------------- # |
|
|
|
# set worker to 4 for shorter dataloader init time |
|
|
|
self.data_num_workers = 4 |
|
|
|
self.input_size = (640, 640) |
|
|
|
self.random_size = (14, 26) |
|
|
|
self.train_ann = "instances_train2017.json" |
|
|
|
self.val_ann = "instances_val2017.json" |
|
|
|
|
|
|
|
# --------------- transform config ----------------- # |
|
|
|
self.degrees = 10.0 |
|
|
|
self.translate = 0.1 |
|
|
|
self.scale = (0.1, 2) |
|
|
|
self.mscale = (0.8, 1.6) |
|
|
|
self.shear = 2.0 |
|
|
|
self.perspective = 0.0 |
|
|
|
self.enable_mixup = True |
|
|
|
|
|
|
|
# -------------- training config --------------------- # |
|
|
|
self.warmup_epochs = 5 |
|
|
|
self.max_epoch = 300 |
|
|
|
self.warmup_lr = 0 |
|
|
|
self.basic_lr_per_img = 0.01 / 64.0 |
|
|
|
self.scheduler = "yoloxwarmcos" |
|
|
|
self.no_aug_epochs = 15 |
|
|
|
self.min_lr_ratio = 0.05 |
|
|
|
self.ema = True |
|
|
|
|
|
|
|
self.weight_decay = 5e-4 |
|
|
|
self.momentum = 0.9 |
|
|
|
self.print_interval = 10 |
|
|
|
self.eval_interval = 10 |
|
|
|
self.exp_name = os.path.split(os.path.realpath(__file__))[1].split(".")[0] |
|
|
|
|
|
|
|
# ----------------- testing config ------------------ # |
|
|
|
self.test_size = (640, 640) |
|
|
|
self.test_conf = 0.001 |
|
|
|
self.nmsthre = 0.65 |
|
|
|
|
|
|
|
def get_model(self): |
|
|
|
from yolox.yolox.models import YOLOPAFPN, YOLOX, YOLOXHead |
|
|
|
@@ -75,4 +38,4 @@ class Exp(BaseExp): |
|
|
|
|
|
|
|
self.model.apply(init_yolo) |
|
|
|
self.model.head.initialize_biases(1e-2) |
|
|
|
return self.model |
|
|
|
return self.model |