Skip to content

Commit e5f4853

Browse files
authored
Add RAM & RAM++ models (#158)
1 parent a3d9196 commit e5f4853

File tree

56 files changed

+9787
-59
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

56 files changed

+9787
-59
lines changed

Cargo.toml

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
[package]
22
name = "usls"
33
edition = "2021"
4-
version = "0.1.7"
4+
version = "0.1.8"
55
rust-version = "1.85"
66
description = "A Rust library integrated with ONNXRuntime, providing a collection of ML models."
77
repository = "https://github.com/jamjamjon/usls"
@@ -141,6 +141,7 @@ trocr = ["tokenizers"]
141141
yolo = ["slsl"]
142142
yoloe = ["yolo", "tokenizers"]
143143
yolop = []
144+
ram = []
144145

145146
# Enable all models
146147
all-models = [
@@ -177,6 +178,7 @@ all-models = [
177178
"trocr",
178179
"yolop",
179180
"mot",
181+
"ram",
180182
]
181183

182184

@@ -375,3 +377,7 @@ required-features = ["yoloe"]
375377
[[example]]
376378
name = "yolop"
377379
required-features = ["yolop"]
380+
381+
[[example]]
382+
name = "ram"
383+
required-features = ["ram"]

README.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -167,6 +167,7 @@ usls = { version = "latest-version", features = [ "cuda" ] }
167167
| [MediaPipe: Selfie-segmentation](https://ai.google.dev/edge/mediapipe/solutions/vision/image_segmenter) | Image Segmentation | `mediapipe-segmenter` | [demo](examples/mediapipe-selfie-segmentation) |
168168
| [Swin2SR](https://github.com/mv-lab/swin2sr) | Image Super-Resolution and Restoration | `swin2sr` | [demo](examples/swin2sr) |
169169
| [APISR](https://github.com/Kiteretsu77/APISR) | Real-World Anime Super-Resolution | `apisr` | [demo](examples/apisr) |
170+
| [RAM & RAM++](https://github.com/xinyu1205/recognize-anything) | Image Tagging | `ram` | [demo](examples/ram) |
170171

171172
</details>
172173

examples/apisr/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
## Quick Start
22

33
```shell
4-
cargo run -r --example apisr
4+
cargo run -r -F apisr --example apisr
55
```
66

77
# Results

examples/ben2/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
## Quick Start
22

33
```shell
4-
cargo run -r --example ben2
4+
cargo run -r -F ben2 --example ben2
55
```

examples/blip/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@ This demo shows how to use [BLIP](https://arxiv.org/abs/2201.12086) to do condit
33
## Quick Start
44

55
```shell
6-
cargo run -r -F cuda --example blip -- --device cuda:0 --source images/dog.jpg --source ./assets/bus.jpg --source images/green-car.jpg
6+
cargo run -r -F blip -F cuda --example blip -- --device cuda:0 --source images/dog.jpg --source ./assets/bus.jpg --source images/green-car.jpg
77
```
88

99
## Results

examples/classifier/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
## Quick Start
22

33
```shell
4-
cargo run -r -F cuda --example classifier -- --device cuda --dtype fp16 --model beit # convnext, fastvit, deit, mobileone
4+
cargo run -r -F image-classifier -F cuda --example classifier -- --device cuda --dtype fp16 --model beit # convnext, fastvit, deit, mobileone
55
```
66

examples/clip/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@ This demo showcases how to use [CLIP](https://github.com/openai/CLIP) to compute
33
## Quick Start
44

55
```shell
6-
cargo run -r -F cuda --example clip -- --device cuda:0 --dtype q4f16
6+
cargo run -r -F clip -F cuda --example clip -- --device cuda:0 --dtype q4f16
77
```
88

99
## Results

examples/d-fine/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
## Quick Start
22

33
```shell
4-
cargo run -r --example d-fine
4+
cargo run -r -F rtdetr --example d-fine
55
```

examples/db/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
## Quick Start
22

33
```shell
4-
cargo run -r -F cuda --example db -- --device cuda --dtype fp16
4+
cargo run -r -F db -F cuda --example db -- --device cuda --dtype fp16
55
```
66

77
## Results

examples/deim/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
## Quick Start
22

33
```shell
4-
cargo run -r --example deim
4+
cargo run -r -F rtdetr --example deim
55
```
66

77

0 commit comments

Comments
 (0)