SikongJueluo/Mini-Nav
1
0
Fork 0
mirror of https://github.com/SikongJueluo/Mini-Nav.git synced 2026-03-12 12:25:32 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat(utils): add feature extraction utilities and tests

Browse Source
This commit is contained in:
SikongJueluo
2026-03-05 20:48:53 +08:00
parent a16b376dd7
commit 5be4709acf
13 changed files with 247 additions and 138 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
mini-nav/main.py
View File

@@ -38,7 +38,7 @@ if __name__ == "__main__":
model_cfg = config.model
processor = cast(
BitImageProcessorFast,
AutoImageProcessor.from_pretrained(model_cfg.name, device_map=device),
AutoImageProcessor.from_pretrained(model_cfg.dino_model, device_map=device),
)
# Load compressor weights if specified in model config
@@ -84,4 +84,6 @@ if __name__ == "__main__":
)
generated_files = synthesizer.generate()
print(f"Generated {len(generated_files)} synthesized images in {dataset_cfg.output_dir}")
print(
f"Generated {len(generated_files)} synthesized images in {dataset_cfg.output_dir}"
)