Upload README.md with huggingface_hub
Browse files
README.md
CHANGED
|
@@ -22,6 +22,8 @@ Checkpoints behind the [CELL-FM CondenSeq demo](https://huggingface.co/spaces/Bo
|
|
| 22 |
| `hpa/vae.bin` | Image VAE at 256x256, the one `hpa/cellfm_seq2img.bin` was trained against | `pretrain_hpa/vae/checkpoint-50000` |
|
| 23 |
| `hpa/cellfm_img2seq.bin` | CELL-FM image-to-sequence model for HPA, 512x512, 3-channel conditioning (includes the ESM-C 600M encoder) | `pretrain_hpa/cellfm_img2seq/checkpoint-60000` |
|
| 24 |
| `hpa/vae_512.bin` | Image VAE at 512x512, the one `hpa/cellfm_img2seq.bin` was trained against — a different model from `hpa/vae.bin`, not a rename | `pretrain_hpa/vae_512/checkpoint-50000` |
|
|
|
|
|
|
|
| 25 |
| `hpa/anchor_cell.npy` | The fixed cell every NLS-screening image is conditioned on: `(3, 256, 256)` float32 in [-1, 1], channels nucleus, ER, microtubules. HPA gene H3C13, cell crop `1194_B2_2_4` | built |
|
| 26 |
| `hpa/anchor_masks.npz` | Two 256x256 boolean masks over that cell, `nucleus` and `cell`; cytoplasm is `cell & ~nucleus` | built |
|
| 27 |
| `hpa/pls_anchor_nls.npz` | The cell PLS generation conditions on for nuclear signals: `cell` `(3, 512, 512)` nucleus/ER/microtubules and `protein` `(1, 512, 512)`, float32 in [-1, 1]. HPA gene PPM1G (Nucleoplasm), crop `392_B9_1_11` | built |
|
|
@@ -36,7 +38,9 @@ Hyperparameters for the CondenSeq models are set in `pipeline.py` in the Space a
|
|
| 36 |
hyperparameters are spelled out in `notebooks/nls_screening.ipynb` and mirror
|
| 37 |
`scripts/cell_fm/evaluate_virtual_staining_hpa_dict.sh`. The img2seq pair mirrors
|
| 38 |
`scripts_local/cell_fm/evaluate_img2seq_hpa_v2.sh`: 512 px, `sample_size` 128,
|
| 39 |
-
`encoder_patch_size` 8, `img_generator_patch_size` 4, 8 attention heads.
|
|
|
|
|
|
|
| 40 |
|
| 41 |
Each generator must be loaded with the VAE it was trained against — pairing
|
| 42 |
`cellfm_img2seq.bin` with the 256 px `vae.bin` gives a latent-size mismatch.
|
|
|
|
| 22 |
| `hpa/vae.bin` | Image VAE at 256x256, the one `hpa/cellfm_seq2img.bin` was trained against | `pretrain_hpa/vae/checkpoint-50000` |
|
| 23 |
| `hpa/cellfm_img2seq.bin` | CELL-FM image-to-sequence model for HPA, 512x512, 3-channel conditioning (includes the ESM-C 600M encoder) | `pretrain_hpa/cellfm_img2seq/checkpoint-60000` |
|
| 24 |
| `hpa/vae_512.bin` | Image VAE at 512x512, the one `hpa/cellfm_img2seq.bin` was trained against — a different model from `hpa/vae.bin`, not a rename | `pretrain_hpa/vae_512/checkpoint-50000` |
|
| 25 |
+
| `opencell/cellfm_seq2img.bin` | CELL-FM virtual-staining generator for OpenCell, 256x256, single nucleus conditioning channel, fine-tuned (includes the ESM-C 600M encoder) | `finetune_opencell/cellfm_vs/checkpoint-100000` |
|
| 26 |
+
| `opencell/vae.bin` | Image VAE at 256x256, the one `opencell/cellfm_seq2img.bin` was trained against — OpenCell-finetuned, so not interchangeable with `hpa/vae.bin` despite the matching shape | `finetune_opencell/vae/checkpoint-50000` |
|
| 27 |
| `hpa/anchor_cell.npy` | The fixed cell every NLS-screening image is conditioned on: `(3, 256, 256)` float32 in [-1, 1], channels nucleus, ER, microtubules. HPA gene H3C13, cell crop `1194_B2_2_4` | built |
|
| 28 |
| `hpa/anchor_masks.npz` | Two 256x256 boolean masks over that cell, `nucleus` and `cell`; cytoplasm is `cell & ~nucleus` | built |
|
| 29 |
| `hpa/pls_anchor_nls.npz` | The cell PLS generation conditions on for nuclear signals: `cell` `(3, 512, 512)` nucleus/ER/microtubules and `protein` `(1, 512, 512)`, float32 in [-1, 1]. HPA gene PPM1G (Nucleoplasm), crop `392_B9_1_11` | built |
|
|
|
|
| 38 |
hyperparameters are spelled out in `notebooks/nls_screening.ipynb` and mirror
|
| 39 |
`scripts/cell_fm/evaluate_virtual_staining_hpa_dict.sh`. The img2seq pair mirrors
|
| 40 |
`scripts_local/cell_fm/evaluate_img2seq_hpa_v2.sh`: 512 px, `sample_size` 128,
|
| 41 |
+
`encoder_patch_size` 8, `img_generator_patch_size` 4, 8 attention heads. The OpenCell pair
|
| 42 |
+
mirrors `scripts/cell_fm/evaluate_virtual_staining_opencell.sh`: 256 px, `sample_size` 64,
|
| 43 |
+
`encoder_patch_size` 4, `img_generator_patch_size` 2, 18 attention heads, `cell_image` `nucl`.
|
| 44 |
|
| 45 |
Each generator must be loaded with the VAE it was trained against — pairing
|
| 46 |
`cellfm_img2seq.bin` with the 256 px `vae.bin` gives a latent-size mismatch.
|