| 85 | |
| 86 | |
| 87 | def build_transform_gen(cfg, is_train): |
| 88 | if is_train: |
| 89 | sample_style = cfg.INPUT.MIN_SIZE_TRAIN_SAMPLING |
| 90 | else: |
| 91 | sample_style = 'choice' |
| 92 | |
| 93 | logger = logging.getLogger(__name__) |
| 94 | tfm_gens = [] |
| 95 | if is_train: |
| 96 | tfm_gens.append(T.RandomFlip(horizontal=True, vertical=False)) |
| 97 | tfm_gens.append(T.ResizeShortestEdge(short_edge_length=cfg.VISDRONE.SHORT_LENGTH, max_size=cfg.VISDRONE.MAX_LENGTH, sample_style=sample_style)) |
| 98 | else: |
| 99 | tfm_gens.append(T.ResizeShortestEdge(short_edge_length=[cfg.VISDRONE.TEST_LENGTH], max_size=cfg.VISDRONE.TEST_LENGTH, sample_style=sample_style)) |
| 100 | |
| 101 | return tfm_gens |
| 102 | |
| 103 | |
| 104 | |