개정판 5cd39521
issue #1366: change file load
DTI_PID/WebServer/symbol_training/src/doftech_dataset.py | ||
---|---|---|
51 | 51 |
img_path = self.img_path_list[index] |
52 | 52 |
self.img_list.append(cv2.imread(img_path)) |
53 | 53 |
|
54 |
if self.is_training==True: |
|
55 |
if self.use_rescale == True: |
|
56 |
new_image_list, new_label_list = self.rescale_func([self.img_list[-1]], [self.anno_list[-1]]) |
|
57 |
|
|
58 |
self.img_list.extend(new_image_list) |
|
59 |
self.anno_list.extend(new_label_list) |
|
60 |
''' |
|
54 | 61 |
if self.is_training==True: |
55 | 62 |
if self.use_rescale == True: |
56 | 63 |
new_image_list, new_label_list = self.rescale_func(self.img_list, self.anno_list) |
57 | 64 |
|
58 | 65 |
self.img_list.extend(new_image_list) |
59 | 66 |
self.anno_list.extend(new_label_list) |
67 |
''' |
|
60 | 68 |
|
61 | 69 |
def rescale_func(self, img_list, anno_list): |
62 | 70 |
#scale_array = [0.7, 0.8, 0.9, 1.0, 1.1, 1.2, 1.3] |
DTI_PID/WebServer/symbol_training/train.py | ||
---|---|---|
88 | 88 |
learning_rate_schedule = {"0": 1e-5, "5": 1e-4, |
89 | 89 |
"80": 1e-5, "110": 1e-6} |
90 | 90 |
|
91 |
training_params = {"batch_size": 1,#opt.batch_size,
|
|
91 |
training_params = {"batch_size": opt.batch_size, |
|
92 | 92 |
"shuffle": True, |
93 | 93 |
"drop_last": True, |
94 | 94 |
"collate_fn": custom_collate_fn} |
내보내기 Unified diff