Skip to content

Commit

Permalink
example output added
Browse files Browse the repository at this point in the history
  • Loading branch information
ahmadjaved97 committed Sep 1, 2024
1 parent 6ec9094 commit e83674e
Showing 1 changed file with 19 additions and 0 deletions.
19 changes: 19 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -45,6 +45,23 @@ python cluster_images.py --image_dataset_path ./images \
--use_feature_dict
```

### Sample Output
- Dataset Used: [Fruit and Vegetable Classification](https://www.kaggle.com/code/abdelrahman16/fruit-and-vegetable-classification/input)
- Number of Clusters: 8
- Model Used: ViT
- Clustering Method: Kmeans
- Output:
<p align="center">
<img src="./output_grids/cluster_0.jpg" alt="Image 1" width="250" height="250">
<img src="./output_grids/cluster_1.jpg" alt="Image 2" width="250" height="250">
<img src="./output_grids/cluster_2.jpg" alt="Image 3" width="250" height= "250">
<img src="./output_grids/cluster_3.jpg" alt="Image 3" width="250" height= "250">
<img src="./output_grids/cluster_4.jpg" alt="Image 3" width="250" height= "250">
<img src="./output_grids/cluster_5.jpg" alt="Image 3" width="250" height= "250">
<img src="./output_grids/cluster_6.jpg" alt="Image 3" width="250" height= "250">
<img src="./output_grids/cluster_7.jpg" alt="Image 3" width="250" height= "250">
</p>

### Argument Description:
- **--image_dataset_path**: Path to the folder containing images.
- --**grid_folder**: Path to save the generated grid images (default: current directory).
Expand All @@ -54,3 +71,5 @@ python cluster_images.py --image_dataset_path ./images \
- --**model**: The model type to use for feature extraction, either vit or resnet.
- --**clustering_method**: The method to use for clustering, either kmeans or gmm.
- --**use_feature_dict**: Use this flag if you want to use an existing feature dictionary instead of recalculating it.


0 comments on commit e83674e

Please sign in to comment.