Skip to content

whs2k/ImageSimilarity

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

32 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

ImageComparison

Determine cosine similarity of a human picture to

Step 1: Download Repo

git clone https://github.com/whs2k/ImageSimilarity.git

Step 2: Setup Environment

conda activate mxnet
conda install jupyter
onda install -c anaconda pandas requests tqdm pil flask pip
pip install --upgrade mxnet gluoncv
conda install -c conda-forge opencv

Step 3: Run Flask App

cd {insert_path_here}/src
python src/0.2.1-whs-invoSho_yolo.py

App Logic

  1. App Initialization (loading of DenseNet Model, YOLO Model, and dataframe of previously vectorized dog photos)
  2. Human Uploads an Image and is saved to the "Static" folder
  3. Yolo Model extractts and crops just the "human" portion of uploaded image
  4. Densenet Model transforms the cropped image to a vector
  5. Cosine Similarity is calculated for tthe uploaded image vector and all previously vectorized dog photos
  6. Top three results are returned

Notes

About

Determine similarity of Images using MXNET

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published