/pyTorchClassification

PyTorch Classification

Primary LanguagePython

Implentmented models:

DictLayer: Dict Layer: A Structured Dictionary Layer[cvprw][2018]

NetVLAD: NetVLAD-CNN architecture for weakly supervised place recognition[cvpr][2016]

Arcface:

Wide&Deep[2016] @ models/location_recommendation.py

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.