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

vd/triplet-loss-cars

Go to file

vd bae1b00bdc added simple example

2022-09-09 23:26:27 +03:00

.model_inference.py.swp

update

2022-09-09 22:29:25 +03:00

classifier_config.json

Update

2022-09-09 21:45:11 +03:00

download-dataset

First commit

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

download-test-videos

Update

2022-09-09 21:45:11 +03:00

embedding_config.json

Update

2022-09-09 21:45:11 +03:00

example.ipynb

added simple example

2022-09-09 23:26:27 +03:00

model_classifier.obj

trained classifier

2022-09-09 23:20:04 +03:00

model_inference.py

update

2022-09-09 22:29:25 +03:00

README.md

Added readme

2022-09-02 03:57:41 +03:00

requirements.txt

update

2022-09-09 22:29:25 +03:00

train-classifier.py

First commit

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

train-embedding.py

Update

2022-09-09 21:45:11 +03:00

yolov5.zip

Update

2022-09-09 21:45:11 +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.