/Cow-Teat-Keratosis-Aug-Stable-Diffusion

Balanced dataset using generated images from Stable Diffusion

Primary LanguageJupyter NotebookGNU General Public License v3.0GPL-3.0

Cow teat data augmentation using DreamBooth's Stable Diffusion a comparison approach

Cow teat dataset contained the following class distribution before Stable Diffusion which shows a very high data imbalance between classes. That's why Data Augmentation was important to generate images for underrepresented classes such as 1, 4, and 3.

Here is the data distribution of the classes: Score_1: 450, Score_2: 491, Score_3: 187, Score_4: 21 Data Distribution, width="100" height="100"

Tested a total of 6 models on both the balanced and unbalanced datasets. The paper can be found here

Model comparison width="100" height="100"

Please reach out to me for any questions or collaboration.