Explorar o código

Fix: SVD decomposition je funkcija, ki se uporablja v Kaubsch-evem algoritmu.

Luka hai 2 meses
pai
achega
b72ffb79e3
Modificáronse 1 ficheiros con 3 adicións e 3 borrados
  1. 3 3
      SeekTransformModule/SeekTransformModule.py

+ 3 - 3
SeekTransformModule/SeekTransformModule.py

@@ -33,7 +33,7 @@ class SeekTransformModuleWidget(ScriptedLoadableModuleWidget):
 
         # Dropdown menu za izbiro metode
         self.rotationMethodComboBox = qt.QComboBox()
-        self.rotationMethodComboBox.addItems(["SVD", "Horn", "Quaternion"])
+        self.rotationMethodComboBox.addItems(["Kabsch", "Horn", "Quaternion"])
         self.layout.addWidget(self.rotationMethodComboBox)
         
         # Load button
@@ -365,7 +365,7 @@ class MyTransformModuleLogic(ScriptedLoadableModuleLogic):
             shNode = slicer.vtkMRMLSubjectHierarchyNode.GetSubjectHierarchyNode(slicer.mrmlScene)
             imageItem = shNode.GetItemByDataNode(volumeNode)
             
-            modality = shNode.GetItemAttribute(imageItem, 'DICOM.Modality')
+            #modality = shNode.GetItemAttribute(imageItem, 'DICOM.Modality')
             #print(modality)
             
             # Check if the volume is loaded into the scene
@@ -431,7 +431,7 @@ class MyTransformModuleLogic(ScriptedLoadableModuleLogic):
                     if selectedMethod == "SVD":
                         chosen_rotation_matrix = compute_Kabsch_rotation(cbct_points, ct_points)
                         chosen_translation_vector = compute_translation(cbct_points, ct_points, chosen_rotation_matrix)
-                        print("\nSVD Method:")
+                        print("\nKabsch Method:")
                         print("Rotation Matrix:\n", chosen_rotation_matrix)
                         print("Translation Vector:\n", chosen_translation_vector)
                     elif selectedMethod == "Horn":