def cropDcm(self): oseg = self.oseg if oseg.data3d is None: self.statusBar().showMessage('No DICOM data!') return self.statusBar().showMessage('Cropping DICOM data...') QApplication.processEvents() pyed = QTSeedEditor(oseg.data3d, mode='crop', voxelSize=oseg.voxelsize_mm) # @TODO mx = self.oseg.viewermax mn = self.oseg.viewermin width = mx - mn # enter = (float(mx)-float(mn)) center = np.average([mx, mn]) logger.debug("window params max %f min %f width, %f center %f" % (mx, mn, width, center)) pyed.changeC(center) pyed.changeW(width) pyed.exec_() crinfo = pyed.getROI() if crinfo is not None: tmpcrinfo = [] for ii in crinfo: tmpcrinfo.append([ii.start, ii.stop]) # seg.data3d = qmisc.crop(oseg.data3d, oseg.crinfo) oseg.crop(tmpcrinfo) self.setLabelText(self.text_dcm_data, self.getDcmInfo()) self.statusBar().showMessage('Ready')
def visualization(self): """ Run viewer with output data3d and segmentation """ try: from pysegbase.seed_editor_qt import QTSeedEditor except: logger.warning("Deprecated of pyseg_base as submodule") from seed_editor_qt import QTSeedEditor from PyQt4.QtGui import QApplication import numpy as np #, QMainWindow app = QApplication(sys.argv) #pyed = QTSeedEditor(self.data3d, contours=(self.segmentation>0)) pyed = QTSeedEditor(self.segmentation) pyed.exec_() #import pdb; pdb.set_trace() #pyed = QTSeedEditor(deletemask, mode='draw') #pyed.exec_() app.exit()
def maskRegion(self): if self.oseg.data3d is None: self.statusBar().showMessage('No DICOM data!') return self.statusBar().showMessage('Mask region...') QApplication.processEvents() pyed = QTSeedEditor( self.oseg.data3d, mode='mask', voxelSize=self.oseg.voxelsize_mm, contours=((self.oseg.segmentation == 0).astype(np.int8)*2) ) pyed.contours_old = pyed.contours.copy() # initial mask set # pyed.masked = np.ones(self.oseg.data3d.shape, np.int8) # pyed.masked = (self.oseg.segmentation == 0).astype(np.int8) mx = self.oseg.viewermax mn = self.oseg.viewermin width = mx - mn # enter = (float(mx)-float(mn)) center = np.average([mx, mn]) logger.debug("window params max %f min %f width, %f center %f" % (mx, mn, width, center)) pyed.changeC(center) pyed.changeW(width) pyed.exec_() self.statusBar().showMessage('Ready')
def test_data_editor(self): """ Funkce provádí změnu vstupních dat - data3d """ #pyed = sed3.sed3(self.data3d, contour = oseg.segmentation) #pyed.show() try: from pysegbase.seed_editor_qt import QTSeedEditor except: logger.warning("Deprecated of pyseg_base as submodule") from seed_editor_qt import QTSeedEditor from PyQt4.QtGui import QApplication import numpy as np im3d = np.random.rand(15,15,15) print ("Select pixels for deletion (it will be setted to 0)") #, QMainWindow app = QApplication(sys.argv) pyed = QTSeedEditor(im3d, mode='draw') pyed.exec_() deletemask = pyed.getSeeds() print ("If it is ok, press 'Return'. If it is wrong, click into image and press 'Return'") # rewrite input data im3d [deletemask != 0] = 0 pyed = QTSeedEditor(im3d) app.exec_() sds = pyed.getSeeds() # if usere select pixel, there will be 1 or 2 in sds self.assertLess(np.max(sds), 1)
def test_viewer_seeds(self): try: from pysegbase.seed_editor_qt import QTSeedEditor except: print("Deprecated of pyseg_base as submodule") from seed_editor_qt import QTSeedEditor from PyQt4.QtGui import QApplication import numpy as np img3d = (np.random.rand(30, 30, 30)*10).astype(np.int16) seeds = (np.zeros(img3d.shape)).astype(np.int8) seeds[3:6, 12:18, 9:16] = 1 seeds[3:6, 19:22, 21:27] = 2 # , QMainWindow app = QApplication(sys.argv) pyed = QTSeedEditor(img3d, seeds=seeds) pyed.exec_() # deletemask = pyed.getSeeds() # import pdb; pdb.set_trace() # pyed = QTSeedEditor(deletemask, mode='draw') # pyed.exec_() app.exit()
def test_data_editor(self): """ Funkce provádí změnu vstupních dat - data3d """ #pyed = sed3.sed3(self.data3d, contour = oseg.segmentation) #pyed.show() try: from pysegbase.seed_editor_qt import QTSeedEditor except: logger.warning("Deprecated of pyseg_base as submodule") from seed_editor_qt import QTSeedEditor from PyQt4.QtGui import QApplication import numpy as np im3d = np.random.rand(15, 15, 15) print("Select pixels for deletion (it will be setted to 0)") #, QMainWindow app = QApplication(sys.argv) pyed = QTSeedEditor(im3d, mode='draw') pyed.exec_() deletemask = pyed.getSeeds() print( "If it is ok, press 'Return'. If it is wrong, click into image and press 'Return'" ) # rewrite input data im3d[deletemask != 0] = 0 pyed = QTSeedEditor(im3d) app.exec_() sds = pyed.getSeeds() # if usere select pixel, there will be 1 or 2 in sds self.assertLess(np.max(sds), 1)
def manualSeg(self): if self.dcm_3Ddata is None: self.statusBar().showMessage('No DICOM data!') return pyed = QTSeedEditor(self.dcm_3Ddata, seeds=self.segmentation_data, mode='draw', voxelSize=self.voxel_sizemm) pyed.exec_() self.segmentation_data = pyed.getSeeds() self.checkSegData()
def maskRegion(self): if self.dcm_3Ddata is None: self.statusBar().showMessage('No DICOM data!') return self.statusBar().showMessage('Mask region...') QApplication.processEvents() pyed = QTSeedEditor(self.dcm_3Ddata, mode='mask', voxelSize=self.voxel_sizemm) pyed.exec_() self.statusBar().showMessage('Ready')
def manualSeg(self): oseg = self.oseg # rint 'ms d3d ', oseg.data3d.shape # rint 'ms seg ', oseg.segmentation.shape # rint 'crinfo ', oseg.crinfo if oseg.data3d is None: self.statusBar().showMessage("No DICOM data!") return sgm = oseg.segmentation.astype(np.uint8) pyed = QTSeedEditor(oseg.data3d, seeds=sgm, mode="draw", voxelSize=oseg.voxelsize_mm, volume_unit="ml") pyed.exec_() oseg.segmentation = pyed.getSeeds() self.oseg.processing_time = datetime.datetime.now() - self.oseg.time_start self.checkSegData("manual seg., ")
def autoSeg(self): if self.dcm_3Ddata is None: self.statusBar().showMessage('No DICOM data!') return igc = pycut.ImageGraphCut(self.dcm_3Ddata, voxelsize=self.voxel_sizemm) pyed = QTSeedEditor(self.dcm_3Ddata, seeds=self.segmentation_seeds, modeFun=igc.interactivity_loop, voxelSize=self.voxel_sizemm) pyed.exec_() self.segmentation_data = pyed.getContours() self.segmentation_seeds = pyed.getSeeds() self.checkSegData()
def cropDcm(self): if self.dcm_3Ddata is None: self.statusBar().showMessage('No DICOM data!') return self.statusBar().showMessage('Cropping DICOM data...') QApplication.processEvents() pyed = QTSeedEditor(self.dcm_3Ddata, mode='crop', voxelSize=self.voxel_sizemm) pyed.exec_() self.dcm_3Ddata = pyed.getImg() self.dcm_offsetmm = pyed.getOffset() self.setLabelText(self.text_dcm_data, self.getDcmInfo()) self.statusBar().showMessage('Ready') self.statusBar().showMessage('Ready')
def interactivity(self, min_val=None, max_val=None): try: from pysegbase.seed_editor_qt import QTSeedEditor except: logger.warning("Deprecated of pyseg_base as submodule") from seed_editor_qt import QTSeedEditor import_gui() logger.debug('interactivity') # if self.edit_data: # self.data3d = self.data_editor(self.data3d) igc = self._interactivity_begin() # from PyQt4.QtCore import pyqtRemoveInputHook # pyqtRemoveInputHook() # import ipdb; ipdb.set_trace() # noqa BREAKPOINT pyed = QTSeedEditor(igc.img, seeds=igc.seeds, modeFun=igc.interactivity_loop, voxelSize=igc.voxelsize, volume_unit='ml') # set window if min_val is None: min_val = np.min(self.data3d) if max_val is None: max_val = np.max(self.data3d) window_c = ((max_val + min_val) / 2) window_w = (max_val - min_val) pyed.changeC(window_c) pyed.changeW(window_w) pyed.exec_() # @TODO někde v igc.interactivity() dochází k přehození nul za jedničy, # tady se to řeší hackem if igc.segmentation is not None: self.segmentation = (igc.segmentation == 0).astype(np.int8) self._interactivity_end(igc)
def btnManualSeg(self): oseg = self.oseg # rint 'ms d3d ', oseg.data3d.shape # rint 'ms seg ', oseg.segmentation.shape # rint 'crinfo ', oseg.crinfo if oseg.data3d is None: self.statusBar().showMessage('No DICOM data!') return sgm = oseg.segmentation.astype(np.uint8) pyed = QTSeedEditor(oseg.data3d, seeds=sgm, mode='draw', voxelSize=oseg.voxelsize_mm, volume_unit='ml') pyed.exec_() oseg.segmentation = pyed.getSeeds() self.oseg.processing_time = \ datetime.datetime.now() - self.oseg.time_start self.checkSegData('manual seg., ')
def maskRegion(self): if self.oseg.data3d is None: self.statusBar().showMessage("No DICOM data!") return self.statusBar().showMessage("Mask region...") QApplication.processEvents() pyed = QTSeedEditor(self.oseg.data3d, mode="mask", voxelSize=self.oseg.voxelsize_mm) mx = self.oseg.viewermax mn = self.oseg.viewermin width = mx - mn # enter = (float(mx)-float(mn)) center = np.average([mx, mn]) logger.debug("window params max %f min %f width, %f center %f" % (mx, mn, width, center)) pyed.changeC(center) pyed.changeW(width) pyed.exec_() self.statusBar().showMessage("Ready")