mirror of
https://github.com/firestar5683/StarPilot.git
synced 2026-11-06 20:43:46 +08:00
firestar5683
26 lines
1.1 KiB
Python
26 lines
1.1 KiB
Python
"""Package the compiled small-model outputs using established Dom model parts."""
|
|
import argparse
|
|
from pathlib import Path
|
|
|
|
|
|
# Resolve this checkout's stdlib-only owner, even for an absolute CLI invocation
|
|
# with no PYTHONPATH or installed workspace package.
|
|
import importlib.util
|
|
_spec = importlib.util.spec_from_file_location("model_file_chunker", Path(__file__).resolve().parents[2] / "openpilot/common/file_chunker.py")
|
|
_chunk_owner = importlib.util.module_from_spec(_spec)
|
|
_spec.loader.exec_module(_chunk_owner)
|
|
package_file = _chunk_owner.package_file
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument('--source', type=Path, required=True)
|
|
parser.add_argument('--destination', type=Path, required=True)
|
|
args = parser.parse_args()
|
|
for name in ('dmonitoring_model_tinygrad.pkl', 'rdf43_driving_tinygrad.pkl'):
|
|
source = (_chunk_owner.materialize_file_chunked(args.source / name)
|
|
if name == 'rdf43_driving_tinygrad.pkl' else args.source / name)
|
|
package_file(source, args.destination / name)
|
|
|
|
if __name__ == '__main__':
|
|
main()
|