search

brian6091 / Dreambooth

Fine-tuning of diffusion models
Apache License 2.0
97 stars 20 forks source link

Augmentation #10

Closed brian6091 closed 1 year ago

brian6091 commented 1 year ago

https://github.com/zhunzhong07/Random-Erasing https://timm.fast.ai/RandomErase https://pytorch.org/vision/main/generated/torchvision.transforms.RandomErasing.html#torchvision.transforms.RandomErasing https://github.com/cjf8899/KeepAugment_Pytorch https://github.com/dvlab-research/GridMask

brian6091 commented 1 year ago

Implemented basic augmentation