czZhang2026 commited on
Commit
924fcea
·
verified ·
1 Parent(s): 4af9cb8

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +49 -1
README.md CHANGED
@@ -1,3 +1,51 @@
1
  ---
2
- license: apache-2.0
 
 
 
 
 
 
 
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
+ library_name: pytorch
3
+ base_model: facebook/VGGT-1B
4
+ datasets:
5
+ - czZhang2026/ViewRecDB-100K
6
+ tags:
7
+ - computer-vision
8
+ - viewpoint-recommendation
9
+ - ai-photography
10
  ---
11
+
12
+ <div align="center">
13
+ <h1>ViewRecNet</h1>
14
+ <p><strong>ViewRec3D: Learning to Recommend 3D Viewpoints for AI Photography</strong></p>
15
+ <p>
16
+ <a href="https://github.com/anonymous1-submission/ViewRec3D/blob/main/ViewRec3D.pdf"><img src="https://img.shields.io/badge/Paper-ViewRec3D-blue" alt="Paper"></a>
17
+ <a href="https://github.com/anonymous1-submission/ViewRec3D"><img src="https://img.shields.io/badge/GitHub-Code-24292f" alt="Code"></a>
18
+ <a href="https://huggingface.co/datasets/czZhang2026/ViewRecDB-100K"><img src="https://img.shields.io/badge/Dataset-ViewRecDB--100K-orange" alt="Dataset"></a>
19
+ </p>
20
+ <p>Chengzhi Zhang<sup>1,*</sup>, Yunzhong Hou<sup>1,*,†</sup>, Zizheng Sun<sup>1</sup>, Ming-Hsuan Yang<sup>2</sup>, Chi Harold Liu<sup>1</sup></p>
21
+ <p><sup>1</sup> Beijing Institute of Technology &nbsp; <sup>2</sup> University of California, Merced</p>
22
+ <p><sup>*</sup> Equal contribution. &nbsp; <sup>†</sup> Corresponding author.</p>
23
+ </div>
24
+
25
+ ## Overview
26
+
27
+ Given a single photograph, ViewRecNet recommends how to move and rotate the camera, adjust the field of view, and choose portrait or landscape orientation to approach an expert composition. The model uses a VGGT backbone and learns from paired views in ViewRecDB-100K with dense camera-ray supervision.
28
+
29
+ ## Released checkpoint
30
+
31
+ `model.safetensors` (4.51 GB) contains the weights from training step 15,625, including the backbone, camera head, orientation head, and ray head. The evaluation code constructs the network and loads these weights directly.
32
+
33
+ ## Quick start
34
+
35
+ From the ViewRec3D code repository, install the project requirements and download the checkpoint:
36
+
37
+ ```bash
38
+ python -m pip install -r requirements.txt
39
+ python -m pip install -U huggingface_hub
40
+ hf download czZhang2026/ViewRecNet model.safetensors --local-dir checkpoints/ViewRecNet
41
+ ```
42
+
43
+ To evaluate on a CUDA GPU, prepare the test split as described in the [dataset card](https://huggingface.co/datasets/czZhang2026/ViewRecDB-100K). Replace the test directory below with its local path. The evaluator reads the checkpoint from `--output_dir` and writes evaluation metrics there.
44
+
45
+ ```bash
46
+ python -m src.evaluate --val_dir /path/to/ViewRecDB-100K/test --output_dir checkpoints/ViewRecNet --use_rays True
47
+ ```
48
+
49
+ ## Acknowledgements
50
+
51
+ This work was supported by the National Natural Science Foundation of China (No. U25A20434).