Checkpoints of ScaleLSD

This repository contains the checkpoints as described in the paper ScaleLSD: Scalable Deep Line Segment Detection Streamlined. For code, see https://github.com/ant-research/scalelsd.

There are two checkpoints of our model, scalelsd-vitbase-v1-train-sa1b.pt and scalelsd-vitbase-v2-train-sa1b.pt.\

The main differences between v1 and v2 models are summarized as follow

Model Name Use of LayerScale Optimizer
scalelsd-vitbase-v1-train-sa1b.pt No ADAM
scalelsd-vitbase-v2-train-sa1b.pt Yes ADAMW

We recommend the users use the v2 model by default because it works much better than v1 (the model evaluated in our paper)

Downloads last month

-

Downloads are not tracked for this model. How to track
Inference Providers NEW
This model isn't deployed by any Inference Provider. ๐Ÿ™‹ Ask for provider support

Space using cherubicxn/scalelsd 1