Implentmented models:

Requirements:

conda install pytorch torchvision cudatoolkit=9.0 -c pytorch
pip install pretrainedmodels imgaug scikit-learn tqdm opencv-python pandas pygeohash cvxpy cvxopt
pip install --upgrade scikit-image

Projects:

1. location recommendation: main_location_company.py

It build the relationship between location(item) and company(user). But each user only buy one item.

2. classification: main_furniture.py

It does classification for furniture.

3. visual localization:

NetVLAD + D2Net for image retrieval and spatial verification.