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>
README.md
CHANGED
|
@@ -1,10 +1,15 @@
|
|
| 1 |
---
|
| 2 |
license: apache-2.0
|
|
|
|
|
|
|
| 3 |
---
|
| 4 |
|
| 5 |
## Checkpoints of ScaleLSD
|
| 6 |
|
| 7 |
-
|
|
|
|
|
|
|
|
|
|
| 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)
|