Releases: okunator/cellseg_models.pytorch
v0.1.29
v0.1.28
v0.1.27
v0.1.26
0.1.26 — 2025-05-07
Removed
- Removed
datamodulesmodule - Removed
datasetsmodule
Refactor
- Refactored the whole model interface to be more user-friendly.
Features
-
Added a new
wsimodule, including:- A
SlideReaderclass to read patches from a WSI slide.- Backends: Openslide, CUCIM
- Adapted the reader class from HistoPrep library. Props to Jopo666
get_sub_gridsfunction to get subgrids from a WSI slide. Can be used to filter the patches. Based on connected components.
- A
-
Added a new torch_datasets module, including:
WSIDatasetInferclass to run inference directly from WSIs.- Adapted the class from HistoPrep library. Props to Jopo666
TrainDatasetH5class to handle training data for the models from a h5 file.TrainDatasetFolderclass to handle training data for the models from img and label folders.
-
Added a new
inference.WsiSegmenter-class to handle the segmentation of WSIs. -
Added a new
wsi.inst_merger.InstMerger-class to handle the merging of instance masks at image boundaries. -
Added
inst2gdfandsem2gdffunctions toutils.vectorizemodule. These functions convert efficiently instance and semantic masks to GeoDataFrame objects. -
Added
FileHandler.to_matandFileHandler.to_gsonsave functions that take in a dictionary of model output masks (output from theInferer-classes) and save it to a .mat or '.feather', '.geojson', '.parquet' files.
Added Dependencies
- Added
libpysaldependency - Added
networkxdependency
Removed Dependencies
- Removed
lightningdependency - Removed
albumentationsdependency
Chore
- Move
FolderDatasetInfertotorch_datasetsmodule
v0.1.25
0.1.25 — 2024-07-05
Features
- Image encoders are imported now only from timm models.
- Add
enc_out_indicesto model classes, to enable selecting which layers to use as the encoder outputs.
Removed
- Removed SAM and DINOv2 original implementation image-encoders from this repo. These can be found from timm models these days.
- Removed
cellseg_models_pytorch.trainingmodule which was left unused after example notebooks were updated.
Examples
- Updated example notebooks.
- Added new example notebooks utilizing UNI foundation model from the MahmoodLab.
- Added new example notebooks utilizing the Prov-GigaPath foundation model from the Microsoft Research.
- NOTE: These examples use the huggingface model hub to load the weights. Permission to use the model weights is required to run these examples.
Chore
- Update timm version to above 1.0.0.
Breaking changes
- Lose support for python 3.9
- The
self.encoderin each model is new, thus, models with trained weights from previous versions of the package will not work with this version.
v0.1.24
0.1.24 — 2023-10-13
Style
- Update the
Ìnferer.infer()-method api to accept arguments related to saving the model outputs.
Features
-
Add
CPP-Net. https://arxiv.org/abs/2102.06867 -
Add option for mixed precision inference
-
Add option to interpolate model outputs to a given size to all of the segmentation models.
-
Add DINOv2 Backbone
-
Add support for
.geojson,.feather,.parquetfile formats when running inference.
Docs
- Add
CPP-Netexample trainng with Pannuke dataset.
Fixes
- Fix resize transformation bug.
v0.1.23
0.1.23 — 2023-08-28
Features
-
add a stem-skip module. (Long skip for the input image resolution feature map)
-
add
UnetTRtransformer encoder wrapper class -
add a new
Encoderwrapper for timm and unetTR based encoders -
Add stem skip support and upsampling block options to all current model architectures
-
Add masking option to all the criterions
-
Add
MAELoss -
Add
BCELoss -
Add base class for transformer based backbones
-
Add
SAM-VitDetimage encoder with support to load pre-trainedSAMweights -
Add
CellVIT-SAMmodel.
Docs
-
Add notebook example on training
Hover-Netwith lightning from scratch. -
Add notebook example on training
StarDistwith lightning from scratch. -
Add notebook example on training
CellPosewith accelerate from scratch. -
Add notebook example on training
OmniPosewith accelerate from scratch. -
Add notebook example on finetuning
CellVIT-SAMwith accelerate.
Fixes
-
Fix current
TimmEncoderto store feature info -
Fix Up block to support transconv and bilinear upsampling and fix data flow issues.
-
Fix
StardistUnetclass to output all the decoder features. -
Fix
Decoder,DecoderStageand long-skip modules to work with up scale factors instead of output dimensions.
v0.1.22
v0.1.21
0.1.21 — 2023-06-12
Features
- Add StrongAugment data augmentation policy to data-loading pipeline: https://arxiv.org/abs/2206.15274
v0.1.20
0.1.20 — 2023-01-13
Fixes
-
Enable only writing folder&hdf5 datasets with only images
-
Enable writing datasets without patching.
-
Add long missing h5 reading utility function to
FileHandler
Features
-
Add hdf5 input file reading to
Infererclasses. -
Add option to write pannuke dataset to h5 db in
PannukeDataModuleandLizardDataModule. -
Add a generic model builder function
get_modeltomodels.__init__.py -
Rewrite segmentation benchmarker. Now it can take in hdf5 datasets.