Skip to content

Predict where the photo was taken from single shot image

Notifications You must be signed in to change notification settings

Hi-king/tokyo_photo2geo

Repository files navigation

Open in Streamlit

tokyo_photo2geo

Predict where the photo was taken from single shot image

Predict your photo

poetry install
poetry run python scripts/predict.py ${your_image_path}

or

notebooks/predict.ipynb

Create your own model

poetry install
KEY=XXXXXX poetry run python scripts/create_dataset.py
poetry run python scripts/train.py \
    --model resnet50 \
    --batch_size 30 \
    --lr 0.00001 \
    --weight_decay 0 \
    --epoch 200

References