(self)
| 10 | |
| 11 | class Exp(MyExp): |
| 12 | def __init__(self): |
| 13 | super(Exp, self).__init__() |
| 14 | self.num_classes = 1 |
| 15 | self.depth = 1.33 |
| 16 | self.width = 1.25 |
| 17 | self.exp_name = os.path.split(os.path.realpath(__file__))[1].split(".")[0] |
| 18 | self.train_ann = "train.json" |
| 19 | self.val_ann = "val_half.json" |
| 20 | self.input_size = (800, 1440) |
| 21 | self.test_size = (800, 1440) |
| 22 | self.random_size = (18, 32) |
| 23 | self.max_epoch = 80 |
| 24 | self.print_interval = 20 |
| 25 | self.eval_interval = 5 |
| 26 | self.test_conf = 0.1 |
| 27 | self.nmsthre = 0.7 |
| 28 | self.no_aug_epochs = 10 |
| 29 | self.basic_lr_per_img = 0.001 / 64.0 |
| 30 | self.warmup_epochs = 1 |
| 31 | |
| 32 | def get_data_loader(self, batch_size, is_distributed, no_aug=False): |
| 33 | from yolox.data import ( |
nothing calls this directly
no outgoing calls
no test coverage detected