okunator/cellseg_models.pytorch: v0.1.23
Description
add a stem-skip module. (Long skip for the input image resolution feature map)
add
UnetTRtransformer encoder wrapper classadd a new
Encoderwrapper for timm and unetTR based encodersAdd stem skip support and upsampling block options to all current model architectures
Add masking option to all the criterions
- Add
MAELoss Add
BCELossAdd base class for transformer based backbones
Add
SAM-VitDetimage encoder with support to load pre-trainedSAMweightsAdd
CellVIT-SAMmodel.
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.
Fix current
TimmEncoderto store feature infoFix 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.
Files
okunator/cellseg_models.pytorch-v0.1.23.zip
Files
(18.4 MB)
| Name | Size | Download all |
|---|---|---|
|
md5:b4997aba0c7a9d693a9fd2c50e7fcb4e
|
18.4 MB | Preview Download |
Additional details
Related works
- Is supplement to
- https://github.com/okunator/cellseg_models.pytorch/tree/v0.1.23 (URL)