Upload PolarAPP SfP and DfP checkpoints
Browse files- DfP/DemNet/DemNet.pth +3 -0
- DfP/FANet/FANet.pth +3 -0
- DfP/TaskNet/TaskNet.pth +3 -0
- README.md +73 -0
- SHA256SUMS.txt +6 -0
- SfP/DemNet/DemNet.pth +3 -0
- SfP/FANet/FANet.pth +3 -0
- SfP/TaskNet/TaskNet.pth +3 -0
DfP/DemNet/DemNet.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:00aa162b962d7ef952cd1c8a0d3bf31ac9bbf9e65d3f76b055bb5c08a64f2bd6
|
| 3 |
+
size 53233207
|
DfP/FANet/FANet.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:20fe6defc1ac37b5d7011bce5b83a43c9e21e6948f36bbb98933a742a7b75a4b
|
| 3 |
+
size 33887292
|
DfP/TaskNet/TaskNet.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e0ec39980e77d85de471e7108df61bacd7a69733750d2bcd37d0fa9382f09096
|
| 3 |
+
size 226812049
|
README.md
ADDED
|
@@ -0,0 +1,73 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
library_name: pytorch
|
| 3 |
+
pipeline_tag: image-to-image
|
| 4 |
+
tags:
|
| 5 |
+
- polarization
|
| 6 |
+
- demosaicking
|
| 7 |
+
- shape-from-polarization
|
| 8 |
+
- reflection-removal
|
| 9 |
+
---
|
| 10 |
+
|
| 11 |
+
# PolarAPP checkpoints
|
| 12 |
+
|
| 13 |
+
This repository contains the released checkpoints for the two downstream tasks in [PolarAPP](https://arxiv.org/abs/2603.23071):
|
| 14 |
+
|
| 15 |
+
- `SfP/`: full-resolution shape from polarization with a TaskNet designed specifically for PolarAPP.
|
| 16 |
+
- `DfP/`: full-resolution de-reflection from polarization with a PolarFree-based TaskNet.
|
| 17 |
+
|
| 18 |
+
## Files
|
| 19 |
+
|
| 20 |
+
```text
|
| 21 |
+
SfP/
|
| 22 |
+
|-- DemNet/DemNet.pth
|
| 23 |
+
|-- TaskNet/TaskNet.pth
|
| 24 |
+
`-- FANet/FANet.pth
|
| 25 |
+
|
| 26 |
+
DfP/
|
| 27 |
+
|-- DemNet/DemNet.pth
|
| 28 |
+
|-- TaskNet/TaskNet.pth
|
| 29 |
+
`-- FANet/FANet.pth
|
| 30 |
+
```
|
| 31 |
+
|
| 32 |
+
`DemNet` reconstructs full-resolution polarization observations. `TaskNet` performs the downstream task. `FANet` supports feature alignment during training.
|
| 33 |
+
|
| 34 |
+
## Download
|
| 35 |
+
|
| 36 |
+
```python
|
| 37 |
+
from huggingface_hub import snapshot_download
|
| 38 |
+
|
| 39 |
+
checkpoint_root = snapshot_download("Roydon728/PolarAPP")
|
| 40 |
+
```
|
| 41 |
+
|
| 42 |
+
Pass the task subdirectory to the corresponding inference command:
|
| 43 |
+
|
| 44 |
+
```bash
|
| 45 |
+
cd SfP
|
| 46 |
+
python infer.py --input-dir ./Datasets/Testsets --ckpt-dir <snapshot>/SfP
|
| 47 |
+
|
| 48 |
+
cd DfP
|
| 49 |
+
python infer.py --input-dir ./Datasets/PolaRGB --ckpt-dir <snapshot>/DfP \
|
| 50 |
+
--polarfree-checkpoint-dir ./experiments/checkpoints/polarfree
|
| 51 |
+
```
|
| 52 |
+
|
| 53 |
+
The DfP workflow loads the PolarFree diffusion-prior files separately using their upstream filenames.
|
| 54 |
+
|
| 55 |
+
## Data
|
| 56 |
+
|
| 57 |
+
- SfP dataset links are distributed through [SfPUEL](https://github.com/YouweiLyu/SfPUEL). Follow the dataset terms and citation guidance from the provider.
|
| 58 |
+
- DfP uses [PolaRGB](https://huggingface.co/datasets/Mingde/PolaRGB), distributed under CC BY-NC 4.0.
|
| 59 |
+
|
| 60 |
+
## Integrity
|
| 61 |
+
|
| 62 |
+
SHA-256 checksums are provided in `SHA256SUMS.txt`.
|
| 63 |
+
|
| 64 |
+
## Citation
|
| 65 |
+
|
| 66 |
+
```bibtex
|
| 67 |
+
@article{luo2026polarapp,
|
| 68 |
+
title = {PolarAPP: Beyond Polarization Demosaicking for Polarimetric Applications},
|
| 69 |
+
author = {Luo, Yidong and Li, Chenggong and Song, Yunfeng and Wang, Ping and Shi, Boxin and Zhang, Junchao and Yuan, Xin},
|
| 70 |
+
journal = {arXiv preprint arXiv:2603.23071},
|
| 71 |
+
year = {2026}
|
| 72 |
+
}
|
| 73 |
+
```
|
SHA256SUMS.txt
ADDED
|
@@ -0,0 +1,6 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
00AA162B962D7EF952CD1C8A0D3BF31AC9BBF9E65D3F76B055BB5C08A64F2BD6 DfP/DemNet/DemNet.pth
|
| 2 |
+
20FE6DEFC1AC37B5D7011BCE5B83A43C9E21E6948F36BBB98933A742A7B75A4B DfP/FANet/FANet.pth
|
| 3 |
+
E0EC39980E77D85DE471E7108DF61BACD7A69733750D2BCD37D0FA9382F09096 DfP/TaskNet/TaskNet.pth
|
| 4 |
+
D7FB0344110E80C196E411DC5406626D287C95A27A78FAFE3DEC9BFCEB6F043A SfP/DemNet/DemNet.pth
|
| 5 |
+
960412138D03EB104FC590302FD5407966D24CB2C0465079136B2AB93D6B5037 SfP/FANet/FANet.pth
|
| 6 |
+
518512B82EFCF53F219FD631E84F759043DA2F11C2ED0AEF88F9F858A6F43B0B SfP/TaskNet/TaskNet.pth
|
SfP/DemNet/DemNet.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d7fb0344110e80c196e411dc5406626d287c95a27a78fafe3dec9bfceb6f043a
|
| 3 |
+
size 46144577
|
SfP/FANet/FANet.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:960412138d03eb104fc590302fd5407966d24cb2c0465079136b2ab93d6b5037
|
| 3 |
+
size 31895813
|
SfP/TaskNet/TaskNet.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:518512b82efcf53f219fd631e84f759043da2f11c2ed0aef88f9f858a6f43b0b
|
| 3 |
+
size 49434735
|