| |
|
| | --- |
| | license: apache-2.0 |
| | tags: |
| | - segmentation |
| | - mamba |
| | - wafer |
| | - electron-microscopy |
| | - tokenunify |
| | --- |
| | |
| | # TokenUnify Models |
| |
|
| | This repository contains TokenUnify models of different sizes trained on wafer electron microscopy data, along with a superhuman baseline model. |
| |
|
| | ## Available Models |
| |
|
| | - **TokenUnify-1B.pth**: 1B parameter TokenUnify model |
| | - **TokenUnify-500M.pth**: 500M parameter TokenUnify model |
| | - **TokenUnify-200M.pth**: 200M parameter TokenUnify model |
| | - **TokenUnify-100M.pth**: 100M parameter TokenUnify model |
| | - **superhuman.pth**: Superhuman baseline model |
| |
|
| | ## Model Details |
| |
|
| | - **Architecture**: TokenUnify (based on Mamba) |
| | - **Training Data**: Wafer electron microscopy images |
| | - **Task**: Image Segmentation |
| | - **Framework**: PyTorch |
| |
|
| | ## Usage |
| |
|
| | ```python |
| | import torch |
| | |
| | # Load a specific model |
| | model_path = "TokenUnify-1B.pth" # or any other model file |
| | checkpoint = torch.load(model_path, map_location='cpu') |
| | |
| | # Your model loading code here |
| | ``` |
| |
|
| | ## Model Sizes |
| |
|
| | | Model | Parameters | File Name | |
| | |-------|------------|-----------| |
| | | TokenUnify Large | 1B | TokenUnify-1B.pth | |
| | | TokenUnify Medium | 500M | TokenUnify-500M.pth | |
| | | TokenUnify Small | 200M | TokenUnify-200M.pth | |
| | | TokenUnify Tiny | 100M | TokenUnify-100M.pth | |
| | | Superhuman Baseline | - | superhuman.pth | |
| |
|
| | ## Citation |
| |
|
| | If you use these models, please cite the relevant paper. |
| | EOF |
| |
|
