SuhZhang/GeoSR-Model

GeoSR Model Zoo

This repository hosts the released checkpoints for GeoSR: Make Geometry Matter for Spatial Reasoning.

Checkpoints

Folder	Branch / task	Notes
GeoSR3D-Model/	static branch	GeoSR checkpoint for static spatial reasoning
GeoSR4D-Model/	dynamic branch	GeoSR checkpoint for dynamic spatial reasoning

The model files are stored exactly as exported from our training directories.

Download

Install the client first if needed:

pip install -U huggingface_hub

Download only the static checkpoint:

python -c "from huggingface_hub import snapshot_download; snapshot_download(repo_id='SuhZhang/GeoSR-Model', local_dir='data/models', allow_patterns=['GeoSR3D-Model/*'])"

Download only the dynamic checkpoint:

python -c "from huggingface_hub import snapshot_download; snapshot_download(repo_id='SuhZhang/GeoSR-Model', local_dir='data/models', allow_patterns=['GeoSR4D-Model/*'])"

Download the whole repository:

git lfs install
git clone https://huggingface.co/SuhZhang/GeoSR-Model

Usage

In the main GeoSR code repository:

• static branch evaluation can use MODEL_PATH=./data/models/GeoSR3D-Model
• dynamic branch evaluation can use GEOSR4D_EVAL_MODEL_PATH=../../../data/models/GeoSR4D-Model

Please refer to the main code repository for full training and evaluation instructions:

• Code: https://github.com/SuhZhang/GeoSR
• Project page: https://suhzhang.github.io/GeoSR/

Citation

@misc{zhang2026geosr,
  title={Make Geometry Matter for Spatial Reasoning},
  author={Shihua Zhang and Qiuhong Shen and Shizun Wang and Tianbo Pan and Xinchao Wang},
  year={2026}
}