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

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +43 -0
README.md ADDED
@@ -0,0 +1,43 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: cc-by-nc-4.0
3
+ tags:
4
+ - glass-surface-detection
5
+ - segmentation
6
+ - pytorch
7
+ datasets:
8
+ - garrying/GSD
9
+ ---
10
+
11
+ # GlassNet — pretrained on GSD
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
+
20
+ ```python
21
+ import torch
22
+ from huggingface_hub import hf_hub_download
23
+ from model import GlassNet
24
+
25
+ ckpt = hf_hub_download("garrying/GSD-GlassNet", "GSD.pth")
26
+ net = GlassNet()
27
+ 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
+
36
+ ```bibtex
37
+ @inproceedings{GSD:2021,
38
+ title = {Rich Context Aggregation with Reflection Prior for Glass Surface Detection},
39
+ author = {Lin, Jiaying and He, Zebang and Lau, Rynson W.H.},
40
+ booktitle = {Proc. CVPR},
41
+ year = {2021}
42
+ }
43
+ ```