Bug fixes (#646)

* "sklearn" import name * Fix render flickering issues * Fix resetting slices on frame change * Lint fix
OpenImaging · Apr 10, 2023 · 4a2d357 · 4a2d357
1 parent de0da65
commit 4a2d357
Show file tree

Hide file tree

Showing 4 changed files with 25 additions and 24 deletions.
diff --git a/miqa/learning/correlator.py b/miqa/learning/correlator.py
@@ -1,7 +1,7 @@
 #!/usr/bin/env python3
 
 import pandas as pd
-from scikit_learn.metrics import confusion_matrix
+from sklearn.metrics import confusion_matrix
 
 df = pd.read_csv('M:/MIQA/data.csv')  # manually converted TRUE/FALSE into 1/0
 print(f'count NaN: {df.isnull().sum().sum()}')

diff --git a/miqa/learning/nn_inference.py b/miqa/learning/nn_inference.py
@@ -4,12 +4,7 @@
 import itk
 import monai
 import numpy as np
-from scikit_learn.metrics import (
-    classification_report,
-    confusion_matrix,
-    mean_squared_error,
-    r2_score,
-)
+from sklearn.metrics import classification_report, confusion_matrix, mean_squared_error, r2_score
 import torch
 from torch.utils.data import DataLoader
 import torchio

diff --git a/miqa/learning/nn_training.py b/miqa/learning/nn_training.py
@@ -21,7 +21,7 @@
 )
 import numpy as np
 import pandas as pd
-from scikit_learn.metrics import confusion_matrix
+from sklearn.metrics import confusion_matrix
 import torch
 from torch.utils.data import DataLoader
 from torch.utils.tensorboard import SummaryWriter

diff --git a/web_client/src/components/VtkViewer.vue b/web_client/src/components/VtkViewer.vue
@@ -104,13 +104,17 @@ export default {
       this.initializeSlice();
       this.initializeView();
     },
-    currentFrame() {
-      this.prepareViewer();
+    currentFrame(oldFrame, newFrame) {
       this.representation.setSlice(this.slice);
-    },
-    currentScan() {
-      this.initializeSlice();
-      this.initializeCamera();
+      this.applyCurrentWindowLevel();
+      this.updateCrosshairs();
+      // use this instead of currentScan watcher
+      // currentScan is computed from currentFrame and technically
+      // will change every time currentFrame has changed
+      if (oldFrame.scan !== newFrame.scan) {
+        this.initializeSlice();
+        this.initializeCamera();
+      }
     },
     showCrosshairs() {
       this.updateCrosshairs();
@@ -150,9 +154,7 @@ export default {
         }
       });
       this.resizeObserver.observe(this.$refs.viewer);
-      const representationProperty = this.representation.getActors()[0].getProperty();
-      representationProperty.setColorWindow(this.currentWindowWidth);
-      representationProperty.setColorLevel(this.currentWindowLevel);
+      this.applyCurrentWindowLevel();
     },
     initializeSlice() {
       if (this.name !== 'default') {
@@ -162,13 +164,12 @@ export default {
     initializeView() {
       this.view.setContainer(this.$refs.viewer);
       fill2DView(this.view);
-      if (this.name !== 'default') {
-        this.modifiedSubscription = this.representation.onModified(() => {
-          if (!this.loadingFrame) {
-            this.slice = this.representation.getSlice();
-          }
-        });
-      }
+      // add scroll interaction to change slice
+      this.view.getInteractor().onMouseWheel(() => {
+        if (!this.loadingFrame) {
+          this.slice = this.representation.getSlice();
+        }
+      });
       // add click interaction to place crosshairs
       this.view.getInteractor().onLeftButtonPress((event) => this.placeCrosshairs(event));
       // remove drag interaction to change window
@@ -206,6 +207,11 @@ export default {
       this.view.resetCamera();
       fill2DView(this.view);
     },
+    applyCurrentWindowLevel() {
+      const representationProperty = this.representation.getActors()[0].getProperty();
+      representationProperty.setColorWindow(this.currentWindowWidth);
+      representationProperty.setColorLevel(this.currentWindowLevel);
+    },
     findClosestColumnToVector(inputVector, matrix) {
       let currClosest = null;
       let currMax = 0;