raab/efishSignalDetector
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

bbox correcter

Browse Source
This commit is contained in:
Till Raab 2023-11-08 16:04:04 +01:00
parent c08ede1e76
commit 397be5c51b
1 changed files with 13 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
correct_bboxes.py
View File

@ -43,24 +43,34 @@ class Bbox_correct_UI(QMainWindow):
self.gridLayout.addWidget(self.win, 0, 0) self.gridLayout.addWidget(self.win, 0, 0)
self.plot_handels[0].getViewBox().invertY(True) self.plot_handels[0].getViewBox().invertY(True)
self.plot_handels[0].getViewBox().setMouseEnabled(x=False, y=False)
img = Image.open(img_path) img = Image.open(img_path)
img_gray = ImageOps.grayscale(img) img_gray = ImageOps.grayscale(img)
img_array = np.array(img_gray).T img_array = np.array(img_gray).T
labels = np.loadtxt(label_path, delimiter=' ') self.labels = np.loadtxt(label_path, delimiter=' ')
self.plot_handels[0].setImage(np.array(img_array)) self.plot_handels[0].setImage(np.array(img_array))
for l in labels: self.ROIs = []
for enu, l in enumerate(self.labels):
x_center, y_center, width, height = l[1:] * img_array.shape[1] x_center, y_center, width, height = l[1:] * img_array.shape[1]
x0, y0, = x_center-width/2, y_center-height/2 x0, y0, = x_center-width/2, y_center-height/2
ROI = pg.RectROI((x0, y0), size=(width, height)) ROI = pg.RectROI((x0, y0), size=(width, height), removable=True)
# ROI.sigRemoveRequested.connect(lambda: )
self.ROIs.append(ROI)
self.plot_widgets[0].addItem(ROI) self.plot_widgets[0].addItem(ROI)
self.plot_widgets[0].setYRange(0, img_array.shape[0], padding=0) self.plot_widgets[0].setYRange(0, img_array.shape[0], padding=0)
self.plot_widgets[0].setXRange(0, img_array.shape[1], padding=0) self.plot_widgets[0].setXRange(0, img_array.shape[1], padding=0)
# def kill_me(self, ROI):
# print('yay')
# print(ROI)
# self.win.removeItem(ROI)
def main_UI(): def main_UI():
app = QApplication(sys.argv) # create application app = QApplication(sys.argv) # create application
img_path = sys.argv[1] img_path = sys.argv[1]