mirror of
https://github.com/SikongJueluo/Mini-Nav.git
synced 2026-03-12 12:25:32 +08:00
refactor(cli): migrate from argparse to typer for command-line interface
This commit is contained in:
25
mini-nav/commands/generate.py
Normal file
25
mini-nav/commands/generate.py
Normal file
@@ -0,0 +1,25 @@
|
||||
import typer
|
||||
|
||||
|
||||
def generate(ctx: typer.Context):
|
||||
from configs import cfg_manager
|
||||
from data_loading.synthesizer import ImageSynthesizer
|
||||
|
||||
config = cfg_manager.get()
|
||||
dataset_cfg = config.dataset
|
||||
|
||||
synthesizer = ImageSynthesizer(
|
||||
dataset_root=dataset_cfg.dataset_root,
|
||||
output_dir=dataset_cfg.output_dir,
|
||||
num_objects_range=dataset_cfg.num_objects_range,
|
||||
num_scenes=dataset_cfg.num_scenes,
|
||||
object_scale_range=dataset_cfg.object_scale_range,
|
||||
rotation_range=dataset_cfg.rotation_range,
|
||||
overlap_threshold=dataset_cfg.overlap_threshold,
|
||||
seed=dataset_cfg.seed,
|
||||
)
|
||||
|
||||
generated_files = synthesizer.generate()
|
||||
typer.echo(
|
||||
f"Generated {len(generated_files)} synthesized images in {dataset_cfg.output_dir}"
|
||||
)
|
||||
Reference in New Issue
Block a user