Add files using upload-large-folder tool
Browse files- README.md +90 -0
- abc/hifi-brep-decoder-4-50.pt +3 -0
- abc/hifi-brep-diffusion-4-50.pt +3 -0
- deepcad/hifi-brep-decoder-4-30.pt +3 -0
- deepcad/hifi-brep-decoder-7-30.pt +3 -0
- deepcad/hifi-brep-diffusion-4-30.pt +3 -0
- deepcad/hifi-brep-diffusion-7-30.pt +3 -0
- training/abc/hifi-brep-vae-4-50-ema.pt +3 -0
- training/deepcad/hifi-brep-vae-4-30-online.pt +3 -0
- training/deepcad/hifi-brep-vae-7-30-online.pt +3 -0
README.md
ADDED
|
@@ -0,0 +1,90 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: apache-2.0
|
| 3 |
+
tags:
|
| 4 |
+
- cad
|
| 5 |
+
- brep
|
| 6 |
+
- 3d-generation
|
| 7 |
+
- diffusion
|
| 8 |
+
- pytorch
|
| 9 |
+
- unconditional-generation
|
| 10 |
+
model_name: HiFi-BRep
|
| 11 |
+
---
|
| 12 |
+
|
| 13 |
+
# HiFi-BRep
|
| 14 |
+
|
| 15 |
+
Checkpoints for **HiFi-BRep: High-Fidelity Latent Representation for Robust B-Rep Generation**.
|
| 16 |
+
|
| 17 |
+
- Code: [1nnoh/HiFi-BRep](https://github.com/1nnoh/HiFi-BRep)
|
| 18 |
+
- Paper: [CVPR 2026](https://openaccess.thecvf.com/content/CVPR2026/papers/Hou_HiFi-BRep_High-Fidelity_Latent_Representation_for_Robust_B-Rep_Generation_CVPR_2026_paper.pdf)
|
| 19 |
+
- HuggingFace: [1nnoh/HiFi-BRep](https://huggingface.co/1nnoh/HiFi-BRep)
|
| 20 |
+
- ModelScope: [innohou/HiFi-BRep](https://www.modelscope.cn/models/innohou/HiFi-BRep)
|
| 21 |
+
|
| 22 |
+
## Inference Checkpoints
|
| 23 |
+
|
| 24 |
+
| Variant | Dataset | Faces | Decoder | Diffusion | State |
|
| 25 |
+
| --- | --- | ---: | --- | --- | --- |
|
| 26 |
+
| `4-30` | DeepCAD | 4–30 | `deepcad/hifi-brep-decoder-4-30.pt` | `deepcad/hifi-brep-diffusion-4-30.pt` | online |
|
| 27 |
+
| `7-30` | DeepCAD | 7–30 | `deepcad/hifi-brep-decoder-7-30.pt` | `deepcad/hifi-brep-diffusion-7-30.pt` | online |
|
| 28 |
+
| `abc-4-50` | ABC | 4–50 | `abc/hifi-brep-decoder-4-50.pt` | `abc/hifi-brep-diffusion-4-50.pt` | EMA |
|
| 29 |
+
|
| 30 |
+
HuggingFace:
|
| 31 |
+
|
| 32 |
+
```bash
|
| 33 |
+
hf download 1nnoh/HiFi-BRep \
|
| 34 |
+
--include "deepcad/*.pt" "abc/*.pt" \
|
| 35 |
+
--local-dir checkpoints
|
| 36 |
+
```
|
| 37 |
+
|
| 38 |
+
ModelScope:
|
| 39 |
+
|
| 40 |
+
```bash
|
| 41 |
+
python -m pip install modelscope-hub==0.1.8
|
| 42 |
+
modelscope download innohou/HiFi-BRep \
|
| 43 |
+
--include "deepcad/*.pt" "abc/*.pt" \
|
| 44 |
+
--local-dir checkpoints
|
| 45 |
+
```
|
| 46 |
+
|
| 47 |
+
## Full-VAE Checkpoints
|
| 48 |
+
|
| 49 |
+
| Variant | Dataset | State | Checkpoint |
|
| 50 |
+
| --- | --- | --- | --- |
|
| 51 |
+
| `4-30` | DeepCAD | online | `training/deepcad/hifi-brep-vae-4-30-online.pt` |
|
| 52 |
+
| `7-30` | DeepCAD | online | `training/deepcad/hifi-brep-vae-7-30-online.pt` |
|
| 53 |
+
| `abc-4-50` | ABC | EMA | `training/abc/hifi-brep-vae-4-50-ema.pt` |
|
| 54 |
+
|
| 55 |
+
HuggingFace:
|
| 56 |
+
|
| 57 |
+
```bash
|
| 58 |
+
hf download 1nnoh/HiFi-BRep \
|
| 59 |
+
--include "training/*/*.pt" \
|
| 60 |
+
--local-dir checkpoints
|
| 61 |
+
```
|
| 62 |
+
|
| 63 |
+
ModelScope:
|
| 64 |
+
|
| 65 |
+
```bash
|
| 66 |
+
python -m pip install modelscope-hub==0.1.8
|
| 67 |
+
modelscope download innohou/HiFi-BRep \
|
| 68 |
+
--include "training/*/*.pt" \
|
| 69 |
+
--local-dir checkpoints
|
| 70 |
+
```
|
| 71 |
+
|
| 72 |
+
## Usage
|
| 73 |
+
|
| 74 |
+
See the repository documentation for [inference](https://github.com/1nnoh/HiFi-BRep#inference-demo), [evaluation](https://github.com/1nnoh/HiFi-BRep/blob/main/docs/EVALUATION.md), [training](https://github.com/1nnoh/HiFi-BRep/blob/main/docs/TRAINING.md), and [data preparation](https://github.com/1nnoh/HiFi-BRep/blob/main/docs/DATA.md).
|
| 75 |
+
|
| 76 |
+
## Citation
|
| 77 |
+
|
| 78 |
+
```bibtex
|
| 79 |
+
@inproceedings{hou2026hifi,
|
| 80 |
+
title={HiFi-BRep: High-Fidelity Latent Representation for Robust B-Rep Generation},
|
| 81 |
+
author={Hou, Junhao and Luo, Chenqi and Wang, Pufan and Lu, Jiaying and Liu, Yusheng and Qin, Feiwei and Fang, Meie and Zhou, Kun},
|
| 82 |
+
booktitle={Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition},
|
| 83 |
+
pages={27199--27208},
|
| 84 |
+
year={2026}
|
| 85 |
+
}
|
| 86 |
+
```
|
| 87 |
+
|
| 88 |
+
## License
|
| 89 |
+
|
| 90 |
+
The code and model artifacts use Apache-2.0. Datasets and third-party dependencies remain subject to their own terms.
|
abc/hifi-brep-decoder-4-50.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:54455bd21ffea304ea950e33ec7ba924816d93ba31bf73c400f68461a5ea36a9
|
| 3 |
+
size 677920525
|
abc/hifi-brep-diffusion-4-50.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a879489cb61ce48ba491ebb62cc5c26124dad5899c16830bedac789d723da87e
|
| 3 |
+
size 773530321
|
deepcad/hifi-brep-decoder-4-30.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:04c593ba8568ac7a1c7dc8c7995611720ad1c01cfafb80d0cc288edc63cfa4fe
|
| 3 |
+
size 677914203
|
deepcad/hifi-brep-decoder-7-30.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d627c5cd528ebba791f9bfbb70022f386bccfec7f4a12926b2089043b2e1cfe6
|
| 3 |
+
size 677914203
|
deepcad/hifi-brep-diffusion-4-30.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:9173b0e0445420107e4ead00af841ccdbe4638487e0f920d6a5733aa17a55270
|
| 3 |
+
size 773526887
|
deepcad/hifi-brep-diffusion-7-30.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:690b866e7a4d21a7f63df28e5045a71c79c101fe94f808944a4271f3b358691a
|
| 3 |
+
size 773526887
|
training/abc/hifi-brep-vae-4-50-ema.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e586b7a8830911d33c7820e59d9c28100aba8557bb0ea685e1247b7ad868712f
|
| 3 |
+
size 1219137434
|
training/deepcad/hifi-brep-vae-4-30-online.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:5552f90304409d09b159571f0d0dbc7c09f9fb5962e92050c91cab5fc17a20ad
|
| 3 |
+
size 1219140536
|
training/deepcad/hifi-brep-vae-7-30-online.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:02a7b1824ac39a34bf298727c5778be353aea4ad9f85ddb99e07eb0b9df7d6b7
|
| 3 |
+
size 1219140536
|