Diff of /3D/configs.py [000000] .. [c9b969]

Switch to side-by-side view

--- a
+++ b/3D/configs.py
@@ -0,0 +1,49 @@
+# image shape
+VOLUME_ROWS = 256
+VOLUME_COLS = 128
+VOLUME_DEPS = 256
+
+# number of classes
+NUM_CLASSES = 4
+
+# patch extract
+PATCH_SIZE = 32
+
+if PATCH_SIZE==64:
+    EXTRACTTION_STEP = 20
+    EXTRACTTION_STEP_CSF = 5
+elif PATCH_SIZE==32:
+    EXTRACTTION_STEP = 10
+    EXTRACTTION_STEP_CSF = 5
+elif PATCH_SIZE==16:
+    EXTRACTTION_STEP = 9
+    EXTRACTTION_STEP_CSF = 4
+    
+# training configs
+UNET_MODEL = 0
+if UNET_MODEL==0:
+    MODEL = 'default'    
+elif UNET_MODEL==1:
+    MODEL = 'reduced'
+elif UNET_MODEL==2:
+    MODEL = 'extended'
+elif UNET_MODEL==3:
+    MODEL = 'extended2'    
+    
+BASE = PATCH_SIZE
+SMOOTH = 1.
+NUM_EPOCHS  = 500
+BATCH_SIZE  = 16
+DEPTH = 5
+EXTRACTION_RECONSTRUCT_STEP = 32
+
+
+if PATCH_SIZE==64:
+    PATIENCE = 10  
+elif PATCH_SIZE==32:
+    PATIENCE = 20  
+elif PATCH_SIZE==16:
+    PATIENCE = 10
+
+# output
+IMAGE_TYPE = '3d_whole_image'