()
| 100 | |
| 101 | |
| 102 | def get_augmentors(): |
| 103 | augs = [] |
| 104 | if args.load_size: |
| 105 | augs.append(imgaug.Resize(args.load_size)) |
| 106 | if args.crop_size: |
| 107 | augs.append(imgaug.CenterCrop(args.crop_size)) |
| 108 | augs.append(imgaug.Resize(args.final_size)) |
| 109 | return augs |
| 110 | |
| 111 | |
| 112 | def get_data(): |