Roydon728 commited on
Commit
39331de
·
verified ·
1 Parent(s): cc50c27

Upload PolarAPP SfP and DfP checkpoints

Browse files
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