From f5b11819599e5ad7041425061e4122acab830f7a Mon Sep 17 00:00:00 2001 From: Christoph Sager Date: Sat, 30 Apr 2022 22:20:34 +0200 Subject: [PATCH] Fix 2D image view (#79) - get file list as string and not as path Closes: #78 --- labelCloud/view/gui.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/labelCloud/view/gui.py b/labelCloud/view/gui.py index 80bfb98..c4b4714 100644 --- a/labelCloud/view/gui.py +++ b/labelCloud/view/gui.py @@ -1,4 +1,5 @@ import logging +import os import re from pathlib import Path from typing import TYPE_CHECKING, Set @@ -434,16 +435,17 @@ def show_2d_image(self): image_folder = config.getpath("FILE", "image_folder") # Look for image files with the name of the point cloud - files_in_image_folder = sorted(image_folder.iterdir()) pcd_name = self.controller.pcd_manager.pcd_path.stem image_file_pattern = re.compile(f"{pcd_name}+(\.(?i:(jpe?g|png|gif|bmp|tiff)))") try: - image_name = next(filter(image_file_pattern.search, files_in_image_folder)) + image_name = next( + filter(image_file_pattern.search, os.listdir(image_folder)) + ) except StopIteration: QMessageBox.information( self, - "No 2d image File", + "No 2D Image File", ( f"Could not find a related image in the image folder ({image_folder}).\n" "Check your path to the folder or if an image for this point cloud exists."