mirror of
https://github.com/20kaushik02/CSE515_MWDB_Project.git
synced 2025-12-06 07:54:07 +00:00
updated README, added requirements.txt
This commit is contained in:
parent
e26108bc3d
commit
d72f2ef2ed
1
Phase 1/Plots/README.md
Normal file
1
Phase 1/Plots/README.md
Normal file
@ -0,0 +1 @@
|
|||||||
|
# Plots generated by task_3.ipynb in show_similar_images()
|
||||||
@ -7,3 +7,25 @@ Getting started: PyTorch, Caltech101 dataset, ResNet50 and similarity measures
|
|||||||
- Color moments - Pearson (faces especially)
|
- Color moments - Pearson (faces especially)
|
||||||
- Histogram of oriented gradients (HOG) - Cosine similarity
|
- Histogram of oriented gradients (HOG) - Cosine similarity
|
||||||
- ResNet50 (avgpool, layer3, fc) - unsatisfactory results for all, simply used euclidean
|
- ResNet50 (avgpool, layer3, fc) - unsatisfactory results for all, simply used euclidean
|
||||||
|
|
||||||
|
## Requirements
|
||||||
|
|
||||||
|
Install packages from requirements.txt (might not be complete)
|
||||||
|
|
||||||
|
## Task 1 - task_1.ipynb
|
||||||
|
|
||||||
|
After installing, run all cells in the notebook. There will be a prompt to give input for image ID. Range is 0 to 8677
|
||||||
|
|
||||||
|
## Task 2 - task_2.ipynb
|
||||||
|
|
||||||
|
Dataset processing and storage to database. Ensure MongoDB server is running, modify connection URI as needed if running on Atlas
|
||||||
|
|
||||||
|
## Task 3 - task_3.ipynb
|
||||||
|
|
||||||
|
Execute all cells till before the "Target images" markdown cell. There will be four prompts to give input for:
|
||||||
|
|
||||||
|
- **Image ID**: integer, 0 to 8677
|
||||||
|
- No. of similar images needed, **k**: positive integer
|
||||||
|
- **Feature model** - one of ["cm", "hog", "avgpool", "layer3", "fc"]
|
||||||
|
- _Note: only hog is applicable for all images. Others cannot be used for grayscale images_
|
||||||
|
- **Similarity/distance measure** - one of ["euclidean", "cosine", "pearson"]
|
||||||
|
|||||||
13
Phase 1/requirements.txt
Normal file
13
Phase 1/requirements.txt
Normal file
@ -0,0 +1,13 @@
|
|||||||
|
numpy
|
||||||
|
scipy
|
||||||
|
scikit-learn
|
||||||
|
scikit-image
|
||||||
|
opencv-python
|
||||||
|
matplotlib
|
||||||
|
torch
|
||||||
|
torchvision
|
||||||
|
pymongo[srv]
|
||||||
|
ipython
|
||||||
|
notebook
|
||||||
|
ipykernel
|
||||||
|
|
||||||
Loading…
x
Reference in New Issue
Block a user