mediapipe icon indicating copy to clipboard operation
mediapipe copied to clipboard

Disable/customise augmentation for the object_detector task

Open DoctorDinosaur opened this issue 3 months ago • 1 comments

MediaPipe Solution (you are using)

Object Detection (Model Maker)

Programming language

Python

Are you willing to contribute it

No

Describe the feature and the current behaviour/state

Currently, while the Image Classification task has "do_data_augmentation", the Object Detection Task does not. The object detection task should be provided this option.

It would also be useful if both could allow for customisation. TF-Vision exposes aug_scale_min, aug_scale_max, aug_rand_hflip, aug_rand_vflip, and aug_type; where aug_type can take a tfm.vision.configs.common.Augmentation (i.e. AutoAugment or RandAugment)

Will this change the current API? How?

No response

Who will benefit with this feature?

No response

Please specify the use cases for this feature

My dataset includes "right arrow" and "left arrow". Random Flipping is enforced and cannot be disabled, without subclassing the object detection class. This means the classes cannot be trained.

Any Other info

No response

DoctorDinosaur avatar May 01 '24 12:05 DoctorDinosaur