199f387b50df1f32ded6ac5d54b68f08f009661b,deeplabcut/refine_training_dataset/refinement.py,MainFrame,nextImage,#MainFrame#Any#,406

Before Change

            img_name = Path(self.img).name

// Plotting
            self.axes.clear()
            self.figure.delaxes(self.figure.axes[1]) // Removes the axes corresponding to the colorbar
            self.figure,self.axes,self.canvas,self.toolbar = self.image_panel.drawplot(self.img,img_name,self.iter,self.index,self.threshold,self.bodyparts,self.colormap,self.preview)

            im = io.imread(self.img)

After Change

            self.figure.delaxes(self.figure.axes[1]) // Removes the axes corresponding to the colorbar
            self.figure,self.axes,self.canvas,self.toolbar = self.image_panel.drawplot(self.img,img_name,self.iter,self.index,self.threshold,self.bodyparts,self.colormap,self.preview,keep_view=self.view_locked)
            im = io.imread(self.img)
            self.axes.callbacks.connect("xlim_changed", self.onZoom)
            self.axes.callbacks.connect("ylim_changed", self.onZoom)
            if np.max(im) == 0:
                msg = wx.MessageBox("Invalid image. Click Yes to remove", "Error!", wx.YES_NO | wx.ICON_WARNING)
                if msg == 2:
                    self.Dataframe = self.Dataframe.drop(self.index[self.iter])
                    self.index = list(self.Dataframe.iloc[:,0].index)
                self.iter = self.iter - 1

                self.img = os.path.join(self.project_path,self.index[self.iter])
                img_name = Path(self.img).name

                self.figure,self.axes,self.canvas,self.toolbar = self.image_panel.drawplot(self.img,img_name,self.iter,self.index,self.threshold,self.bodyparts,self.colormap,self.preview,keep_view=self.view_locked)
                self.axes.callbacks.connect("xlim_changed", self.onZoom)
                self.axes.callbacks.connect("ylim_changed", self.onZoom)
            MainFrame.plot(self,self.img)
        else:
            self.next.Enable(False)

Instances