vd/triplet-loss-cars - triplet-loss-cars - DriveCast git repo

vd/triplet-loss-cars

Go to file

vd 996a86c982 Added readme

2022-09-02 03:56:34 +03:00

classifier_config.json

First commit

2022-09-02 03:48:26 +03:00

download-dataset

First commit

2022-09-02 03:48:26 +03:00

embedding_config.json

First commit

2022-09-02 03:48:26 +03:00

README.md

Added readme

2022-09-02 03:56:34 +03:00

requirements.txt

First commit

2022-09-02 03:48:26 +03:00

train-classifier.py

First commit

2022-09-02 03:48:26 +03:00

train-embedding.py

First commit

2022-09-02 03:48:26 +03:00

README.md

First download existing dataset with download-dataset. Run train-embedding.py to train embedding model with embedding_config.json. If visualize is set to true will create loss graph and PCA+TSNE visualization. Run train-classifier.py to train classification (kNN) model with classifier_config.json