cherubicxn nielsr HF Staff commited on
Commit
fb12333
·
verified ·
1 Parent(s): cbc017c

Add pipeline tag, library name and link to code (#1)

Browse files

- Add pipeline tag, library name and link to code (bd478b4b030264b55fac7d323a11c034cca0a733)


Co-authored-by: Niels Rogge <nielsr@users.noreply.huggingface.co>

Files changed (1) hide show
  1. README.md +7 -2
README.md CHANGED
@@ -1,10 +1,15 @@
1
  ---
2
  license: apache-2.0
 
 
3
  ---
4
 
5
  ## Checkpoints of ScaleLSD
6
 
7
- There are two checkpoints of our model, [scalelsd-vitbase-v1-train-sa1b.pt](https://huggingface.co/cherubicxn/scalelsd/blob/main/scalelsd-vitbase-v1-train-sa1b.pt) and [scalelsd-vitbase-v2-train-sa1b.pt](https://huggingface.co/cherubicxn/scalelsd/blob/main/scalelsd-vitbase-v2-train-sa1b.pt).
 
 
 
8
 
9
  The main differences between v1 and v2 models are summarized as follow
10
 
@@ -13,4 +18,4 @@ The main differences between v1 and v2 models are summarized as follow
13
  | [scalelsd-vitbase-v1-train-sa1b.pt](https://huggingface.co/cherubicxn/scalelsd/blob/main/scalelsd-vitbase-v1-train-sa1b.pt) | No | ADAM |
14
  | [scalelsd-vitbase-v2-train-sa1b.pt](https://huggingface.co/cherubicxn/scalelsd/blob/main/scalelsd-vitbase-v2-train-sa1b.pt) | Yes | ADAMW |
15
 
16
- We recommend the users use the v2 model by default because it works much better than v1 (the model evaluated in our paper)
 
1
  ---
2
  license: apache-2.0
3
+ library_name: pytorch
4
+ pipeline_tag: image-feature-extraction
5
  ---
6
 
7
  ## Checkpoints of ScaleLSD
8
 
9
+ This repository contains the checkpoints as described in the paper [ScaleLSD: Scalable Deep Line Segment Detection Streamlined](https://huggingface.co/papers/2506.09369).
10
+ For code, see https://github.com/ant-research/scalelsd.
11
+
12
+ There are two checkpoints of our model, [scalelsd-vitbase-v1-train-sa1b.pt](https://huggingface.co/cherubicxn/scalelsd/blob/main/scalelsd-vitbase-v1-train-sa1b.pt) and [scalelsd-vitbase-v2-train-sa1b.pt](https://huggingface.co/cherubicxn/scalelsd/blob/main/scalelsd-vitbase-v2-train-sa1b.pt).\
13
 
14
  The main differences between v1 and v2 models are summarized as follow
15
 
 
18
  | [scalelsd-vitbase-v1-train-sa1b.pt](https://huggingface.co/cherubicxn/scalelsd/blob/main/scalelsd-vitbase-v1-train-sa1b.pt) | No | ADAM |
19
  | [scalelsd-vitbase-v2-train-sa1b.pt](https://huggingface.co/cherubicxn/scalelsd/blob/main/scalelsd-vitbase-v2-train-sa1b.pt) | Yes | ADAMW |
20
 
21
+ We recommend the users use the v2 model by default because it works much better than v1 (the model evaluated in our paper)