GitHub - crasanders/ChatGPT-Categorization

Install the following packages -

pip install timm
pip install open_clip_torch
pip install scipy

To extract CLIP embeddings -

python embed_clip.py --image_folder="PATH_TO_360_ROCK_IMAGES"

To extract embeddings from (torch)vision models -

python embed_vision.py --image_folder="PATH_TO_360_ROCK_IMAGES"

The image_embeddings.zip contains the generated embeddings from all the models in .csv and .mat format.

Name		Name	Last commit message	Last commit date
data		data
output		output
rocks_embeddings_generation		rocks_embeddings_generation
.gitattributes		.gitattributes
.gitignore		.gitignore
README.md		README.md
chatgpt_mds.txt		chatgpt_mds.txt
chatgpt_ratings.csv		chatgpt_ratings.csv
chatgpt_similarities.csv		chatgpt_similarities.csv
create_batch_file.py		create_batch_file.py
dimension_ratings.ipynb		dimension_ratings.ipynb
get_similarities.py		get_similarities.py
get_similarities_with_dimension_ratings.ipynb		get_similarities_with_dimension_ratings.ipynb
mds1x2.png		mds1x2.png
mds3x4.png		mds3x4.png
mds5x6.png		mds5x6.png
mds7x8.png		mds7x8.png
mds_analysis.ipynb		mds_analysis.ipynb
number_map.json		number_map.json
plot_mds.ipynb		plot_mds.ipynb
process_batch_output.ipynb		process_batch_output.ipynb
reliability.ipynb		reliability.ipynb
run_batches.py		run_batches.py
similartiy_ratings.ipynb		similartiy_ratings.ipynb
upload_batch_file.ipynb		upload_batch_file.ipynb

Navigation Menu