Skip to content

Ivaldivi/DATA515-Image-Classification

Repository files navigation

DATA515-Landmark-Classification

Team members: Sarah Innis, Anthony Nguyen, Annie Staker, Izzy Valdivia

Build/Test Workflow Coverage Status

Deployed Website:

Cascadia Classifier is available here: Cascadia Classifier

The Project Type:

Web-based tool

Questions of Interest:

  • How accurately can we classify landmarks?
  • Can we distinguish between similar-appearing landmarks (e.g. same area)?
  • Can we identify landmarks through different environments and lighting conditions?
  • Can we identify landmarks if given a low-quality image input?
  • Can we identify landmarks that we have not trained our model upon?
  • Can we provide further information about a landmark?

Goal for project output:

Create a multiclass image classification tool that can correctly identify landmarks (tentatively within Washington state) if given an image.

Data Sources:

  • Images of Landmarks across the world, provided by Google
    • More than 4 million labeled landmark photos
  • Wikimedia
    • (specifically, the Wikimedia link for a given landmark)
    • Determine location information about a landmark
    • Provide additional information about a landmark
  • User-taken Photos of Landmarks in Washington
    • Use to validate model & demonstrate model accuracy and precision

Download Data:

  • Note: unzip these files in walandmarks/data
  • Data for WikiMedia Scraping
  • Data For Training Our Model

How To Use Conda Environment:

In order to create a conda environment from the current environment.yml file, run the following line:
conda env create -f environment.yml

To activate the environment run:
conda activate landmarks_classification_env

Streamlit:

After configuring an environment, run the Streamlit app using this command:
python -m streamlit run walandmarks/ui/Home.py

Unit testing:

Our unit tests can be found at walandmarks/test.

To run all unit tests:
python -m unittest discover

To see code coverage:
coverage run -m unittest discover

Pylint:

This repo uses PyLint as its linter. To run pylint:
PYTHONPATH=. pylint walandmarks

About

A multi-class image classification tool to identify landmarks in Washington State

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Contributors 4

  •  
  •  
  •  
  •