default pass

Kapoorlabs-CAPED · Aug 22, 2023 · fb6045c · fb6045c
1 parent 412033a
commit fb6045c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/src/oneat/NEATModels/neat_densevollnet.py b/src/oneat/NEATModels/neat_densevollnet.py
@@ -314,8 +314,8 @@ def predict(self,
            print(f'zero padded image shape ${self.image.shape}')
            self.second_pass_predict()
         if self.remove_markers == None:
+           self.pad_width = (self.config['imagey'], self.config['imagex']) 
            self.image = np.zeros([self.originalimage.shape[0], self.originalimage.shape[1],  self.originalimage.shape[2] + self.pad_width[0], self.originalimage.shape[3] + self.pad_width[1] ]) 
-           self.pad_width = (self.config['imagey'], self.config['imagex'])
            for i in range(self.originalimage.shape[0]):
               self.image[i,:] = pad_timelapse(self.originalimage[i,:], self.pad_width)
            self.default_pass_predict() 

diff --git a/src/oneat/NEATModels/neat_vollnet.py b/src/oneat/NEATModels/neat_vollnet.py
@@ -311,8 +311,9 @@ def predict(self,
            print(f'zero padded image shape ${self.image.shape}')
            self.second_pass_predict()
         if self.remove_markers == None:
+           self.pad_width = (self.config['imagey'], self.config['imagex']) 
            self.image = np.zeros([self.originalimage.shape[0], self.originalimage.shape[1],  self.originalimage.shape[2] + self.pad_width[0], self.originalimage.shape[3] + self.pad_width[1] ]) 
-           self.pad_width = (self.config['imagey'], self.config['imagex'])
+
            for i in range(self.originalimage.shape[0]):
               self.image[i,:] = pad_timelapse(self.originalimage[i,:], self.pad_width) 
            self.image = self.originalimage