malte-soe / landmark_recognition_challenge

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Code style: black

Preparation

Create Environment

  • Navigate into this repository
  • Execute following command: conda env create -f environment.yml
  • Activate the environment: conda activate landmark-recognition

Download Datasets

  • Download one or more datasets from Datasets
    • It doesn't matter where you save them
    • Don't rename the downloaded csv. They should be named train or test.
  • Navigate into this repository
  • Activate the environment
  • Execute following command: python scripts/download_dataset.py --name={FOLDER_NAME} --csv={PATH_TO_DOWNLOADED_CSV}
    • You have to download train and test images separately
    • Images are saved in ./data/{FOLDER_NAME}/{CSV_NAME}/{ID}.jpg
    • For testing purpose you can also download only the first N images with passing --num {N}

Directories

  • ./data/ contains the datasets.
  • ./evaluation/ contains the evaluation artifacts, such as evaluated metrics.
  • ./tensorboard/ contains the tensorboard logs.
  • ./log/ contains logs, such as logged stdout.
  • ./experiments/ contains experiment scripts.
    • experiment scripts must be named after the following structure: exp_{ID}_{NAME}.py

Datasets

Experiments

Start experiments with python main.py {ID} {FLAGS} Checkpoints, evaluation artifacts, logs are stored in sub directories named after the experiment and passed flags.

00 Test

Only for testing purpose.

01 ResNet50 trained with triplet loss

Crop input image into 5 sub images and extract features on each one. Uses triplet loss. Does not work yet!

About


Languages

Language:Python 100.0%