extract_frames.py

import os
import sys
from pathlib import Path

from PySide6 import QtWidgets
from PySide6.QtWidgets import QApplication

from frame_extractor.widgets.frame_extractor import FrameExtractor


bundle_dir = getattr(sys, '_MEIPASS', os.getcwd())
res_folder = Path(os.path.abspath(os.path.join(bundle_dir, 'res')))
app_icon_file = res_folder / "logo.png"

app = QApplication(sys.argv)
app.setApplicationName("Labell3R Frame Extractor")

dialog = QtWidgets.QFileDialog()
dialog.setFileMode(QtWidgets.QFileDialog.FileMode.ExistingFile)
dialog.setNameFilter("Config File (*.yaml)")
dialog.setWindowTitle("Select Config File")
dialog.setAcceptMode(QtWidgets.QFileDialog.AcceptMode.AcceptOpen)

if dialog.exec():
    file_path = dialog.selectedFiles()[0]
    config_file = Path(file_path)
else:
    exit(1)

image_folder = config_file.parent / "images"
context_folder = config_file.parent / "context"

image_folder.mkdir(parents=True, exist_ok=True)
context_folder.mkdir(parents=True, exist_ok=True)

window = FrameExtractor(image_folder, context_folder)
window.show()

sys.exit(app.exec())