garrying commited on
Commit
71607d0
·
verified ·
1 Parent(s): 004806b

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +28 -4
README.md CHANGED
@@ -12,8 +12,16 @@ datasets:
12
 
13
  Pretrained checkpoint for **GlassNet** from the CVPR 2021 paper:
14
 
15
- > **Rich Context Aggregation with Reflection Prior for Glass Surface Detection**
16
- > Jiaying Lin, Zebang He, Rynson W.H. Lau
 
 
 
 
 
 
 
 
17
 
18
  ## Usage
19
 
@@ -28,8 +36,20 @@ net.load_state_dict(torch.load(ckpt, map_location="cpu"))
28
  net.eval()
29
  ```
30
 
31
- See the [dataset page](https://huggingface.co/datasets/garrying/GSD) and the
32
- [project homepage](https://jiaying.link/cvpr2021-gsd/) for full inference instructions.
 
 
 
 
 
 
 
 
 
 
 
 
33
 
34
  ## Citation
35
 
@@ -41,3 +61,7 @@ See the [dataset page](https://huggingface.co/datasets/garrying/GSD) and the
41
  year = {2021}
42
  }
43
  ```
 
 
 
 
 
12
 
13
  Pretrained checkpoint for **GlassNet** from the CVPR 2021 paper:
14
 
15
+ > **Rich Context Aggregation with Reflection Prior for Glass Surface Detection**
16
+ > Jiaying Lin, Zebang He, Rynson W.H. Lau
17
+ > *Proceedings of CVPR 2021*
18
+ > Project page: https://jiaying.link/cvpr2021-gsd/
19
+
20
+ ## Files
21
+
22
+ | File | Description |
23
+ |------|-------------|
24
+ | `GSD.pth` | GlassNet weights trained on the GSD training split |
25
 
26
  ## Usage
27
 
 
36
  net.eval()
37
  ```
38
 
39
+ For full inference code (data loading, CRF post-processing, saving outputs) see `infer.py` in the original release.
40
+
41
+ ## Model Architecture
42
+
43
+ **GlassNet** uses a ResNeXt-101 backbone with:
44
+
45
+ - **DenseContrastModule** — multi-scale dilated convolutions (rates 1/2/4/8) with pairwise feature subtraction to capture cross-context contrast
46
+ - **SELayer** — grouped squeeze-and-excitation for context-aware channel reweighting
47
+ - **RefNet** — a lightweight U-Net-style decoder that jointly predicts the binary glass mask and reconstructs the reflection image as auxiliary output
48
+ - **CRF post-processing** — dense CRF refinement of predicted masks at inference time
49
+
50
+ ## Dataset
51
+
52
+ The GSD dataset is available at [garrying/GSD](https://huggingface.co/datasets/garrying/GSD).
53
 
54
  ## Citation
55
 
 
61
  year = {2021}
62
  }
63
  ```
64
+
65
+ ## Contact
66
+
67
+ jiayinlin5-c@my.cityu.edu.hk