For all main contributors, please check contributing.
This repository is dedicated to research and development of AI-based place recognition.
Clone this GitHub repository:
git clone https://github.com/SKHU-Adventure/ai-place-recognition.git
cd ai-place-recognition
The main branch works with CUDA 12.1, CUDNN 8.9.2, NCCL 2.18.3 with Python 3.8. Refer to a document for pre-setting and then install requirements:
pip install -r requirements.txt
- Prepare dataset for training:
For public datasets, you may refer to following links.
-
Create a directory for your experiment (e.g.,
experiments/sample
) -
Write a setup.ini file in the directory (e.g.,
experiments/sample/setup.ini
). -
Run:
python3 train.py [EXPERIMENT_DIR]
where [EXPERIMENT_DIR]
is the directory created above.
Main contributors:
- Mujae Park,
mujae9837[at]gmail.com
- Younguk Jeon,
jayiuk987[at]gmail.com
- Heeju Cha,
chaheeju01[at]gmail.com
- Younah Kim,
kkiwiio[at]gmail.com
Advisior:
- Sangyun Lee,
sylee[at]skhu.ac.kr