Rename crate: brainiac-rs -> brainiac
Browse files
README.md
CHANGED
|
@@ -9,7 +9,7 @@ tags:
|
|
| 9 |
- vit
|
| 10 |
- foundation-model
|
| 11 |
- medical-imaging
|
| 12 |
-
library_name: brainiac
|
| 13 |
pipeline_tag: feature-extraction
|
| 14 |
---
|
| 15 |
|
|
@@ -52,7 +52,7 @@ The backbone can be fine-tuned for:
|
|
| 52 |
- **Time-to-stroke prediction** (regression)
|
| 53 |
- **Tumor segmentation** (UNETR decoder)
|
| 54 |
|
| 55 |
-
## Usage with brainiac
|
| 56 |
|
| 57 |
```bash
|
| 58 |
cargo run --release --bin infer -- \
|
|
@@ -61,7 +61,7 @@ cargo run --release --bin infer -- \
|
|
| 61 |
```
|
| 62 |
|
| 63 |
```rust
|
| 64 |
-
use
|
| 65 |
|
| 66 |
let (encoder, _) = BrainiacEncoder::<B>::load(
|
| 67 |
"backbone.safetensors", None,
|
|
|
|
| 9 |
- vit
|
| 10 |
- foundation-model
|
| 11 |
- medical-imaging
|
| 12 |
+
library_name: brainiac
|
| 13 |
pipeline_tag: feature-extraction
|
| 14 |
---
|
| 15 |
|
|
|
|
| 52 |
- **Time-to-stroke prediction** (regression)
|
| 53 |
- **Tumor segmentation** (UNETR decoder)
|
| 54 |
|
| 55 |
+
## Usage with brainiac (Rust)
|
| 56 |
|
| 57 |
```bash
|
| 58 |
cargo run --release --bin infer -- \
|
|
|
|
| 61 |
```
|
| 62 |
|
| 63 |
```rust
|
| 64 |
+
use brainiac::{BrainiacEncoder, TaskType};
|
| 65 |
|
| 66 |
let (encoder, _) = BrainiacEncoder::<B>::load(
|
| 67 |
"backbone.safetensors", None,
|