Esempio n. 1
0
    def on_clicked_select_images_button(self):
        selected_category = self.__selected_dataset_category()
        if selected_category is None:
            print('TODO: disable to select other items')
            return

        ext_filter = '画像ファイル(*.jpg *.jpeg *.png *.gif *.bmp)'
        source_image_names = QFileDialog.getOpenFileNames(
            caption='データセットに取り込む',
            filter=ext_filter,
            directory=Project.latest_dataset_image_path())[0]
        Project.save_latest_dataset_image_path(
            os.path.dirname(source_image_names[0]))
        if source_image_names:
            for source_image_name in source_image_names:
                try:
                    # TODO: specify correct camera number
                    destination = Dataset.generate_image_path(
                        category=selected_category,
                        cam_number=0,
                        file_extension=Path(source_image_name).suffix)
                    shutil.copyfile(source_image_name, destination)
                except shutil.SameFileError:
                    print("TODO: fix destination")