Skip to content

Commit 026ba3f

Browse files
committed
Add fliplr method to ROIs class for left-right flipping of ROIs
1 parent 117b106 commit 026ba3f

File tree

1 file changed

+20
-0
lines changed

1 file changed

+20
-0
lines changed

face_rhythm/rois.py

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -329,7 +329,27 @@ def plot_rois(self, image=None, **kwargs_imshow):
329329
## show figure
330330
plt.show()
331331
return fig, ax
332+
333+
def fliplr(self):
334+
"""
335+
Flip the ROIs left-right. In place
336+
"""
337+
if hasattr(self, 'exampleImage'):
338+
if self.exampleImage is not None:
339+
self.exampleImage = np.fliplr(self.exampleImage)
332340

341+
if hasattr(self, 'mask_images'):
342+
if self.mask_images is not None:
343+
self.mask_images = {k: np.fliplr(m) for k, m in self.mask_images.items()}
344+
345+
if hasattr(self, 'roi_points'):
346+
if self.roi_points is not None:
347+
for k, p in self.roi_points.items():
348+
self.roi_points[k][:, 0] = self.img_hw[1] - p[:, 0]
349+
350+
if hasattr(self, 'point_positions'):
351+
if self.point_positions is not None:
352+
self.point_positions[:, 0] = self.img_hw[1] - self.point_positions[:, 0]
333353

334354

335355
class _Select_ROI:

0 commit comments

Comments
 (0)