ykcheungab
66 lines
2.1 KiB
Python
66 lines
2.1 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
UV-unroll finger tactile pipeline with unified Qt dashboard.
|
|
|
|
Where to customize the GUI:
|
|
Layout (sizes, proportions) -> examples/gui/layout.py <-- edit here
|
|
Colors / QSS -> examples/gui/styles.py
|
|
Widget rendering logic -> examples/gui/widgets/
|
|
Window behavior -> examples/gui/dashboard.py
|
|
"""
|
|
|
|
import argparse
|
|
import os
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
os.environ.setdefault("OPENCV_VIDEOIO_MSMF_ENABLE_HW_TRANSFORMS", "0")
|
|
|
|
_EXAMPLES_DIR = Path(__file__).resolve().parent
|
|
if str(_EXAMPLES_DIR) not in sys.path:
|
|
sys.path.insert(0, str(_EXAMPLES_DIR))
|
|
|
|
from gui import run_dashboard
|
|
from pipeline import PipelineConfig
|
|
from viz import DEFAULT_FINGER_VIZ_POSE
|
|
|
|
|
|
def _parse_args():
|
|
parser = argparse.ArgumentParser(
|
|
description="UV-unroll finger tactile pipeline with unified Qt dashboard"
|
|
)
|
|
parser.add_argument("--video", "-v", default="0", help="Camera index or video file path")
|
|
parser.add_argument("--config", "-c", default="./config/dd02-ov.json", help="Orisys SDK config")
|
|
parser.add_argument("--cal", default="", help="Optional calibration file path")
|
|
parser.add_argument("--verbose", action="store_true", help="SDK verbose logs")
|
|
parser.add_argument(
|
|
"--mesh-map",
|
|
choices=["flow", "depth"],
|
|
default="depth",
|
|
help="Color the 3D mesh from flow magnitude or contact depth.",
|
|
)
|
|
parser.add_argument("--threshold", type=float, default=None, help="Mesh/arrow threshold")
|
|
parser.add_argument("--no-3d", action="store_true", help="Disable 3D mesh view")
|
|
parser.add_argument(
|
|
"--viz-pose",
|
|
default=DEFAULT_FINGER_VIZ_POSE,
|
|
help="world_to_cam.npy for the default view; pass 'none' to skip",
|
|
)
|
|
parser.add_argument(
|
|
"--cutoff",
|
|
type=float,
|
|
default=5.0,
|
|
help="Low-pass cutoff for force curves (Hz)",
|
|
)
|
|
return parser.parse_args()
|
|
|
|
|
|
def main():
|
|
args = _parse_args()
|
|
config = PipelineConfig.from_args(args)
|
|
sys.exit(run_dashboard(config))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|