Add Tensorflow based local detection (#95)

Closes: #12 Reviewed-on: Georgi_Lab/ALVINN_f7#95
2024-02-14 19:42:32 -07:00
parent 7757d2348a
commit 94d4bbb979
13 changed files with 719 additions and 51 deletions
--- a/src/models/thorax_tfwm/classes.json
+++ b/src/models/thorax_tfwm/classes.json
@@ -0,0 +1,12 @@
+[
+  "Right lung",
+  "Diaphragm",
+  "Heart",
+  "Caudal vena cava",
+  "Cranial vena cava",
+  "Phrenic nerve",
+  "Trachea",
+  "Vagus nerve",
+  "Left Lung",
+  "Aorta"
+]
--- a/src/models/thorax_tfwm/group1-shard1of7.bin
+++ b/src/models/thorax_tfwm/group1-shard1of7.bin
--- a/src/models/thorax_tfwm/group1-shard2of7.bin
+++ b/src/models/thorax_tfwm/group1-shard2of7.bin
--- a/src/models/thorax_tfwm/group1-shard3of7.bin
+++ b/src/models/thorax_tfwm/group1-shard3of7.bin
--- a/src/models/thorax_tfwm/group1-shard4of7.bin
+++ b/src/models/thorax_tfwm/group1-shard4of7.bin
--- a/src/models/thorax_tfwm/group1-shard5of7.bin
+++ b/src/models/thorax_tfwm/group1-shard5of7.bin
--- a/src/models/thorax_tfwm/group1-shard6of7.bin
+++ b/src/models/thorax_tfwm/group1-shard6of7.bin
--- a/src/models/thorax_tfwm/group1-shard7of7.bin
+++ b/src/models/thorax_tfwm/group1-shard7of7.bin
--- a/src/models/thorax_tfwm/model.json
+++ b/src/models/thorax_tfwm/model.json