Metadata-Version: 2.4
Name: lung-cancer-astra-weights
Version: 0.1.0
Summary: 3D CT image backbone weights (I3ResNet-152) for lung-cancer-astra, delivered as shard packages
License: MIT
Requires-Python: >=3.8
Description-Content-Type: text/markdown
Requires-Dist: lung-cancer-astra-weights-merlin-p1==0.1.0
Requires-Dist: lung-cancer-astra-weights-merlin-p2==0.1.0
Requires-Dist: lung-cancer-astra-weights-merlin-p3==0.1.0
Provides-Extra: astra
Requires-Dist: lung-cancer-astra-weights-astra-p1==0.1.0; extra == "astra"
Requires-Dist: lung-cancer-astra-weights-astra-p2==0.1.0; extra == "astra"
Requires-Dist: lung-cancer-astra-weights-astra-p3==0.1.0; extra == "astra"

# lung-cancer-astra-weights

3D CT image backbone weights (Inflated 3D ResNet-152) used by
[`lung-cancer-astra`](https://pypi.org/project/lung-cancer-astra/).

PyPI limits a single file to 100 MB and each weight file is 225 MiB, so every
weight is published as three shard packages. This package depends on them and
joins the shards on first use, verifying the SHA-256 of the result.

```bash
pip install lung-cancer-astra-weights            # merlin backbone
pip install "lung-cancer-astra-weights[astra]"   # also the astra backbone
```
```python
import lung_cancer_astra_weights as w
w.path("merlin")   # -> path of the joined weight file (cached)
w.verify("astra")
```

The joined files are written to `$LCA_WEIGHTS_DIR` or `~/.cache/lung_cancer_astra`.

| Weight | Size | Source | License |
|---|---|---|---|
| `merlin` | 235 MB | `stanfordmimi/Merlin` | MIT (Stanford MIMI Lab) |
| `astra` | 235 MB | `zzzwzh/Astra_CTRG` | MIT |

Both files hold the image backbone only (930 tensors, 117.5 M parameters, bfloat16);
the text encoder, the perceiver and the language model are not included.
