Update model card metadata, links and usage instructions
Browse filesHi, I'm Niels from the community science team at Hugging Face. I've updated your model card to improve its visibility and usability:
- Updated placeholder badges for arXiv and ICML 2026 with the actual project links.
- Added `base_model` and `datasets` metadata to link this repository to the FLUX model and the SA-Z dataset.
- Included the CLI and Streamlit inference instructions from your GitHub README to provide a "Quick Start" guide for users.
- Linked the paper to the Hugging Face papers page.
README.md
CHANGED
|
@@ -1,14 +1,18 @@
|
|
| 1 |
---
|
| 2 |
-
|
|
|
|
| 3 |
library_name: diffusers
|
|
|
|
| 4 |
pipeline_tag: text-to-image
|
|
|
|
|
|
|
|
|
|
| 5 |
tags:
|
| 6 |
- occlusion
|
| 7 |
- layout
|
| 8 |
- flux
|
| 9 |
-
language:
|
| 10 |
-
- en
|
| 11 |
---
|
|
|
|
| 12 |
<table align="center" cellspacing="0" cellpadding="0" style="margin:0 auto;">
|
| 13 |
<tr>
|
| 14 |
<td valign="middle" style="padding-right:10px;">
|
|
@@ -23,9 +27,9 @@ language:
|
|
| 23 |
|
| 24 |
<div align="center" style="margin-top:14px;">
|
| 25 |
<a href='https://henghuiding.com/OcclusionFormer/'><img src='https://img.shields.io/badge/Project-Page-Green'></a>
|
| 26 |
-
<a href='https://icml.cc/
|
| 27 |
-
<a href='https://arxiv.org/'><img src='https://img.shields.io/badge/arXiv-
|
| 28 |
-
<a href='https://huggingface.co/'><img src='https://img.shields.io/badge/%F0%9F%A4%97%
|
| 29 |
<a href='https://huggingface.co/datasets/FudanCVL/SA-Z'><img src='https://img.shields.io/badge/%F0%9F%A4%97%20SA--Z-Hugging Face%20-orange'></a>
|
| 30 |
</div>
|
| 31 |
<p align="center" style="margin:4px 0 0 0;">
|
|
@@ -36,7 +40,12 @@ language:
|
|
| 36 |
<p align="center" style="margin:1px 0 0 0; font-size:1.48em; font-weight:900; color:#ff6a00;">ICML 2026</p>
|
| 37 |
<p align="center" style="margin:1px 0 0 0; font-size:1.08em; color:#6b7280;"><em>✉ Corresponding Author</em></p>
|
| 38 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 39 |
## 🔥 News
|
|
|
|
| 40 |
- [2026/05/18] Release **inference code**, **model weights** and **SA-Z dataset**.
|
| 41 |
- [2026/05/18] Release **OcclusionFormer open-source package** in this repository.
|
| 42 |
- [2026/4/30] OcclusionFormer is accepted to **ICML 2026**.
|
|
@@ -85,9 +94,7 @@ pip install --upgrade -r requirements.txt
|
|
| 85 |
|
| 86 |
3. Download checkpoint
|
| 87 |
|
| 88 |
-
```
|
| 89 |
-
https://huggingface.co/FudanCVL/OcclusionFormer
|
| 90 |
-
```
|
| 91 |
|
| 92 |
4. Run Streamlit demo
|
| 93 |
|
|
@@ -107,18 +114,6 @@ python inference_occlusionformer.py \
|
|
| 107 |
--overwrite
|
| 108 |
```
|
| 109 |
|
| 110 |
-
Batch inference with a directory of JSON layouts:
|
| 111 |
-
|
| 112 |
-
```bash
|
| 113 |
-
python inference_occlusionformer.py \
|
| 114 |
-
--model_path /path/to/FLUX.1-dev \
|
| 115 |
-
--ckpt_path /path/to/occlusionformer_checkpoint_dir \
|
| 116 |
-
--layout_dir ./examples \
|
| 117 |
-
--output_dir ./outputs_occlusionformer \
|
| 118 |
-
--enable_layout \
|
| 119 |
-
--overwrite
|
| 120 |
-
```
|
| 121 |
-
|
| 122 |
---
|
| 123 |
## ✅ TODO
|
| 124 |
|
|
|
|
| 1 |
---
|
| 2 |
+
language:
|
| 3 |
+
- en
|
| 4 |
library_name: diffusers
|
| 5 |
+
license: mit
|
| 6 |
pipeline_tag: text-to-image
|
| 7 |
+
base_model: black-forest-labs/FLUX.1-dev
|
| 8 |
+
datasets:
|
| 9 |
+
- FudanCVL/SA-Z
|
| 10 |
tags:
|
| 11 |
- occlusion
|
| 12 |
- layout
|
| 13 |
- flux
|
|
|
|
|
|
|
| 14 |
---
|
| 15 |
+
|
| 16 |
<table align="center" cellspacing="0" cellpadding="0" style="margin:0 auto;">
|
| 17 |
<tr>
|
| 18 |
<td valign="middle" style="padding-right:10px;">
|
|
|
|
| 27 |
|
| 28 |
<div align="center" style="margin-top:14px;">
|
| 29 |
<a href='https://henghuiding.com/OcclusionFormer/'><img src='https://img.shields.io/badge/Project-Page-Green'></a>
|
| 30 |
+
<a href='https://icml.cc/virtual/2026/poster/66159'><img src='https://img.shields.io/badge/ICML-2026-blue'></a>
|
| 31 |
+
<a href='https://arxiv.org/abs/2605.21343'><img src='https://img.shields.io/badge/arXiv-2605.21343-b31b1b.svg'></a>
|
| 32 |
+
<a href='https://huggingface.co/papers/2605.21343'><img src='https://img.shields.io/badge/%F0%9F%A4%97%20Paper-Hugging Face-orange'></a>
|
| 33 |
<a href='https://huggingface.co/datasets/FudanCVL/SA-Z'><img src='https://img.shields.io/badge/%F0%9F%A4%97%20SA--Z-Hugging Face%20-orange'></a>
|
| 34 |
</div>
|
| 35 |
<p align="center" style="margin:4px 0 0 0;">
|
|
|
|
| 40 |
<p align="center" style="margin:1px 0 0 0; font-size:1.48em; font-weight:900; color:#ff6a00;">ICML 2026</p>
|
| 41 |
<p align="center" style="margin:1px 0 0 0; font-size:1.08em; color:#6b7280;"><em>✉ Corresponding Author</em></p>
|
| 42 |
|
| 43 |
+
This repository contains the weights for **OcclusionFormer**, as presented in the paper [OcclusionFormer: Arranging Z-Order for Layout-Grounded Image Generation](https://huggingface.co/papers/2605.21343).
|
| 44 |
+
|
| 45 |
+
The official code is available at: [https://github.com/FudanCVL/OcclusionFormer](https://github.com/FudanCVL/OcclusionFormer)
|
| 46 |
+
|
| 47 |
## 🔥 News
|
| 48 |
+
- [2026/05/21] Release **OcclusionFormer** on [ArXiv](https://arxiv.org/abs/2605.21343)!
|
| 49 |
- [2026/05/18] Release **inference code**, **model weights** and **SA-Z dataset**.
|
| 50 |
- [2026/05/18] Release **OcclusionFormer open-source package** in this repository.
|
| 51 |
- [2026/4/30] OcclusionFormer is accepted to **ICML 2026**.
|
|
|
|
| 94 |
|
| 95 |
3. Download checkpoint
|
| 96 |
|
| 97 |
+
Weights can be found in this repository. Download `occlusionformer` to `./ckpt`.
|
|
|
|
|
|
|
| 98 |
|
| 99 |
4. Run Streamlit demo
|
| 100 |
|
|
|
|
| 114 |
--overwrite
|
| 115 |
```
|
| 116 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 117 |
---
|
| 118 |
## ✅ TODO
|
| 119 |
|