def get_next_image(self):
images, idx = self.get_files()
idx = (idx + 1) % len(images)
io.load_image(self, filename=images[idx])
def dropEvent(self, event):
files = [u.toLocalFile() for u in event.mimeData().urls()]
if os.path.splitext(files[0])[-1] == '.npy':
io.load_seg(self, filename=files[0])
else:
io.load_image(self, filename=files[0])
def __init__(self, image=None):
super(MainW, self).__init__()
pg.setConfigOptions(imageAxisOrder="row-major")
self.setGeometry(50, 50, 1200, 1000)
self.setWindowTitle("cellpose")
self.cp_path = os.path.dirname(os.path.realpath(__file__))
app_icon = QtGui.QIcon()
icon_path = os.path.abspath(os.path.join(self.cp_path,
"logo/logo.png"))
app_icon.addFile(icon_path, QtCore.QSize(16, 16))
app_icon.addFile(icon_path, QtCore.QSize(24, 24))
app_icon.addFile(icon_path, QtCore.QSize(32, 32))
app_icon.addFile(icon_path, QtCore.QSize(48, 48))
self.setWindowIcon(app_icon)
menus.mainmenu(self)
menus.editmenu(self)
menus.helpmenu(self)
self.setStyleSheet("QMainWindow {background: 'black';}")
self.stylePressed = ("QPushButton {Text-align: left; "
"background-color: rgb(100,50,100); "
"border-color: white;"
"color:white;}")
self.styleUnpressed = ("QPushButton {Text-align: left; "
"background-color: rgb(50,50,50); "
"border-color: white;"
"color:white;}")
self.styleInactive = ("QPushButton {Text-align: left; "
"background-color: rgb(30,30,30); "
"border-color: white;"
"color:rgb(80,80,80);}")
self.loaded = False
# ---- MAIN WIDGET LAYOUT ---- #
self.cwidget = QtGui.QWidget(self)
self.l0 = QtGui.QGridLayout()
self.cwidget.setLayout(self.l0)
self.setCentralWidget(self.cwidget)
self.l0.setVerticalSpacing(4)
self.imask = 0
b = self.make_buttons()
# ---- drawing area ---- #
self.win = pg.GraphicsLayoutWidget()
self.l0.addWidget(self.win, 0, 3, b, 20)
self.win.scene().sigMouseClicked.connect(self.plot_clicked)
self.win.scene().sigMouseMoved.connect(self.mouse_moved)
self.make_viewbox()
bwrmap = make_bwr()
self.bwr = bwrmap.getLookupTable(start=0.0, stop=255.0, alpha=False)
self.cmap = []
for i in range(3):
self.cmap.append(
make_cmap(i).getLookupTable(start=0.0, stop=255.0,
alpha=False))
self.colormap = (
plt.get_cmap('gist_ncar')(np.linspace(0.0, .9, 1000)) *
255).astype(np.uint8)
self.reset()
self.is_stack = True # always loading images of same FOV
# if called with image, load it
if image is not None:
self.filename = image
io.load_image(self, self.filename)
self.setAcceptDrops(True)
self.win.show()
self.show()