A package to segmentation tumor on CT and MRI
Project description
MEDSEGLIB
This package designed for segmentation of oncology by CT and MRI images.
Models
Currently, the models DeepLabV3(bb: resnet50), DeepLabV3(bb: resnet101) are used. During the training, these models were used, as well as UNet and TransUnet, but DeepLabV3 showed the best results (dice and iou).
Supported types of research
- Brain
- Liver
How to use
import numpy as np
from megseglib.sdk import Model
ct = np.zeros((1, 3, 512, 512), dtype=np.single)
model = Model(model_name="deeplab_v3_resnet101", model_type="liver")
result = model.forward(ct)
# or
result = model(ct)
print(type(result), result.shape) # <class 'numpy.ndarray'> (1, 1, 512, 512) - segmentation mask
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distributions
No source distribution files available for this release.See tutorial on generating distribution archives.
Built Distribution
Close
Hashes for medseglib-1.0.0-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 124bf5d6026aea016c283de649a495dfab3998ab87a355c700437873b6e5d825 |
|
MD5 | 46d13e7f61ad94874668632a191fe8b9 |
|
BLAKE2b-256 | f2af47a138f3fa22d9d2d3f844c49ceafded8ef8bef17106f08fbed4b749372f |