First commit

This commit is contained in:
vd
2022-09-02 03:48:26 +03:00
commit 57dc12b305
8 changed files with 456 additions and 0 deletions

6
classifier_config.json Normal file
View File

@@ -0,0 +1,6 @@
{
"n_neighbors": 10,
"classes" : ["black_red", "green_orange", "yellow_grey"],
"path_to_dataset": "./triplet_dataset",
"embedding_model": "./model_embedding.pt"
}