First commit

2022-09-02 03:48:26 +03:00
commit 57dc12b305
8 changed files with 456 additions and 0 deletions
--- a/embedding_config.json
+++ b/embedding_config.json
@@ -0,0 +1,15 @@
+{
+	"epochs": 4,
+	"embedding_dims": 128,
+	"batch_size": 32,
+	"classes" : ["black_red", "green_orange", "yellow_grey"],
+	"lr": 0.001,
+	"triplet-loss-margin": 1.0,
+	"triplet-loss-p": 2,
+	"train_size": 0.9,
+	"augmentations": true,	
+	"env": "",
+	"path_to_dataset": "./triplet_dataset",
+	"visualize": true,
+	"pca_components": 16
+}