Make YOLOX Nano conversion wrapper usable from the Hub checkout

source/compile-yolox-nano.py

@@ -6,11 +6,14 @@ import runpy
 
 parser = argparse.ArgumentParser(description=__doc__)
 parser.add_argument("--zoo", type=Path, required=True)
+parser.add_argument("--onnx", type=Path, required=True, help="YOLOX Nano ONNX from the pinned Rockchip exporter")
+parser.add_argument("--output-dir", type=Path, required=True)
 args = parser.parse_args()
-root = Path(__file__).resolve().parent
-official = runpy.run_path(str(root.parent.parent / "convert-official.py"))
+source_dir = Path(__file__).resolve().parent
+root = args.output_dir.resolve()
+official = runpy.run_path(str(source_dir / "convert-official.py"))
 source = "https://github.com/airockchip/YOLOX/blob/main/README_rkopt.md"
-model = root / "onnx/yolox/yolox_nano.onnx"
+model = args.onnx.resolve()
 asset = ("yolox", "yolox_nano", model, source)
 with ThreadPoolExecutor(4) as pool:
     jobs = [pool.submit(official["convert"], root, args.zoo.resolve(), asset, chip, precision)