How to use Augmentation in TLT Python?

Hey I m trying to use rotation spatial augmentation in the TLT training pipeline but It is not working

spatial_augmentation {
hflip_probability: 0.5
vflip_probability: 0.0
zoom_min: 1.0
zoom_max: 1.0
translate_max_x: 8.0
translate_max_y: 8.0
rotation_angle_max: 30.0
rotation_units: "degrees"

  }

Error: google.protobuf.text_format.ParseError: 48:5 : Message type “AugmentationConfig.SpatialAugmentation” has no field named “rotation_angle_max”.

I tried the way mentioned here in the documentation which tells us to use it like this

rotation_config{
angle: 5.0
units: "degrees"
}

But it is also not working, and I see that the code is not open source.
So please tell me the right FIELD name which can be used for using the rotation Augmentation in the detection pipeline.

Thanks

https://docs.nvidia.com/metropolis/TLT/tlt-user-guide/text/object_detection/offline_data_augmentation.html#id3

+1

Please note that it is only compatible with the tool augment. See Offline Data Augmentation — Transfer Learning Toolkit 3.0 documentation

In order to circumvent this and generate a dataset with the required augmentations and give control to the user, TLT provides an offline augmentation tool called augment .

Please download latest jupyter notebook.
see
https://docs.nvidia.com/metropolis/TLT/tlt-user-guide/text/quickstart/deepstream_integration.html#download-jupyter-notebook

wget --content-disposition https://api.ngc.nvidia.com/v2/resources/nvidia/tlt_cv_samples/versions/v1.0.2/zip -O tlt_cv_samples_v1.0.2.zip
unzip -u tlt_cv_samples_v1.0.2.zip -d ./tlt_cv_samples_v1.0.2 && rm -rf tlt_cv_samples_v1.0.2.zip && cd ./tlt_cv_samples_v1.0.2

then, go into tlt_cv_samples_v1.0.2/augment.
There is augment.ipynb and its specs folder.

# Spec file for tlt-augment.
spatial_config{
rotation_config{
angle: 10.0
units: “degrees”
}
}
dataset_config{
image_path: “image_2”
label_path: “label_2”
}
color_config{
hue_saturation_config{
hue_rotation_angle: 5.0
saturation_shift: 1.0
}
}
output_image_width: 1248
output_image_height: 384
output_image_channel: 3
image_extension: “.png”