프로젝트

일반

사용자정보

개정판 5cd39521

ID5cd395212377ee2ae0eaf9e83d830292b41b8bbb
상위 b2027eec
하위 48e59210

함의성이(가) 4년 이상 전에 추가함

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

클립보드 이미지 추가 (최대 크기: 500 MB)