TIO-IKIM / CellViT

CellViT: Vision Transformers for Precise Cell Segmentation and Classification
https://doi.org/10.1016/j.media.2024.103143
Other
217 stars 33 forks source link

fix: :bug: Improving cell detection performance, fixed bugs, extendin… #8

Closed FabianHoerst closed 1 year ago

FabianHoerst commented 1 year ago

…g documentation, add mixed precision

Improving runtime for inference by using an improved structure. Postprocessing of cell detection is speed up by a factor of x100. Also fixed a small bug regarding double cell entries in the overlap area. Adding mixed precision as optional training option. Adding more documentation

Double cell detection in cell_detection.py and speed is increased by a large margin