Re #40: add Slicer module for reconstructing segmentation from Pytorc…

…h model
SlicerIGT · May 27, 2023 · 67ee881 · 67ee881
1 parent 363e16f
commit 67ee881
Show file tree

Hide file tree

Showing 5 changed files with 1,476 additions and 0 deletions.
diff --git a/SlicerExtension/LiveUltrasoundAi/TorchSequenceSegmentation/CMakeLists.txt b/SlicerExtension/LiveUltrasoundAi/TorchSequenceSegmentation/CMakeLists.txt
@@ -0,0 +1,31 @@
+#-----------------------------------------------------------------------------
+set(MODULE_NAME TorchSequenceSegmentation)
+
+#-----------------------------------------------------------------------------
+set(MODULE_PYTHON_SCRIPTS
+  ${MODULE_NAME}.py
+  )
+
+set(MODULE_PYTHON_RESOURCES
+  Resources/Icons/${MODULE_NAME}.png
+  Resources/UI/${MODULE_NAME}.ui
+  )
+
+#-----------------------------------------------------------------------------
+slicerMacroBuildScriptedModule(
+  NAME ${MODULE_NAME}
+  SCRIPTS ${MODULE_PYTHON_SCRIPTS}
+  RESOURCES ${MODULE_PYTHON_RESOURCES}
+  WITH_GENERIC_TESTS
+  )
+
+#-----------------------------------------------------------------------------
+if(BUILD_TESTING)
+
+  # Register the unittest subclass in the main script as a ctest.
+  # Note that the test will also be available at runtime.
+  slicer_add_python_unittest(SCRIPT ${MODULE_NAME}.py)
+
+  # Additional build-time testing
+  add_subdirectory(Testing)
+endif()