Skip to content

Can I disable specific augmentations? #764

Answered by rwightman
sk-zk asked this question in Q&A

You must be logged in to vote

@sk-zk I have not added a clean interface to do that, but there is a reasonable hack, you can modify the private _RAND_TRANSFORMS or _RAND_INCREASING_TRANSFORMS (if you specify inc1` in the rand aug string) before you create the data pipeline and remove the translates.

https://github.com/rwightman/pytorch-image-models/blob/master/timm/data/auto_augment.py#L542-L580

Replies: 1 comment

You must be logged in to vote
0 replies
Answer selected by sk-zk
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Category
Q&A
Labels
None yet
2 participants