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

6 lines
314 B
Markdown

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`