Custom TTA modifications for detectron2

Implement both HFlip and VFlip as TTA transforms, add support for RotatedBoxes

source

DatasetMapperTTAFlip

 DatasetMapperTTAFlip (min_sizes:List[int], max_size:int, flip:bool)

Implement test-time augmentation for detection data. Modified to implement both horizontal and vertical flip It is a callable which takes a dataset dict from a detection dataset, and returns a list of dataset dicts where the images are augmented from the input image by the transformations defined in the config. This is used for test-time augmentation.


source

TTAPredictor

 TTAPredictor (cfg)

DefaultPredictor that implements TTA