Files
triplet-loss-cars/README.md
2022-09-02 03:57:41 +03:00

321 B

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.