(datadir, isTrain=True)
| 158 | |
| 159 | |
| 160 | def get_data(datadir, isTrain=True): |
| 161 | if isTrain: |
| 162 | augs = [ |
| 163 | imgaug.Resize(int(SHAPE * 1.12)), |
| 164 | imgaug.RandomCrop(SHAPE), |
| 165 | imgaug.Flip(horiz=True), |
| 166 | ] |
| 167 | else: |
| 168 | augs = [imgaug.Resize(SHAPE)] |
| 169 | |
| 170 | def get_image_pairs(dir1, dir2): |
| 171 | def get_df(dir): |
| 172 | files = sorted(glob.glob(os.path.join(dir, '*.jpg'))) |
| 173 | df = ImageFromFile(files, channel=3, shuffle=isTrain) |
| 174 | return AugmentImageComponent(df, augs) |
| 175 | return JoinData([get_df(dir1), get_df(dir2)]) |
| 176 | |
| 177 | names = ['trainA', 'trainB'] if isTrain else ['testA', 'testB'] |
| 178 | df = get_image_pairs(*[os.path.join(datadir, n) for n in names]) |
| 179 | df = BatchData(df, BATCH if isTrain else TEST_BATCH) |
| 180 | df = MultiProcessRunnerZMQ(df, 2 if isTrain else 1) |
| 181 | return df |
| 182 | |
| 183 | |
| 184 | class VisualizeTestSet(Callback): |
no test coverage detected