Merge pull request #22 from ibois-epfl/ttool_classification_new_weights

UPDATE: updated the weights according to zenodo
ibois-epfl · Jan 25, 2024 · cdd006c · cdd006c
2 parents 6ff7f34 + 8d5e3bd
commit cdd006c
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 9 deletions.
diff --git a/ai/torchscripts/efficientnet.pt b/ai/torchscripts/efficientnet.pt
diff --git a/ai/torchscripts/labels.txt b/ai/torchscripts/labels.txt
@@ -1,13 +1,11 @@
-brad_point_drill_bit_20_150
-spade_drill_bit_25_150
 chain_swordsaw_blade_200
-saber_saw_blade_makita
-self_feeding_bit_50_90
-auger_drill_bit_20_450
-twist_drill_bit_32_165
+chain_saw_blade_f_250
+spade_drill_bit_25_150
 auger_drill_bit_20_235
-auger_drill_bit_20_400
+twist_drill_bit_32_165
 self_feeding_bit_40_90
+saber_saw_blade_makita
 auger_drill_bit_34_235
-chain_saw_blade_f_250
-circular_saw_blade_makita_190
+brad_point_drill_bit_20_150
+circular_saw_blade_makita_190
+self_feeding_bit_50_90