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

314 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