search

TIO-IKIM / CellViT

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

Error report for cellvit_cpp_net.py #49

Closed won5830 closed 5 months ago

won5830 commented 5 months ago

For line 358 ~363 dist_map_features = self._forward_upsample( z0, z1, z2, z3, z4, self.dist_decoder) type_map_features = self._forward_upsample( z0, z1, z2, z3, z4, self.dist_decoder) dist_map_features & type_map_features both upsamples from self.dist_decoder. Shouldn't the latter one be self.nuclei_type_maps_decoder ?

FabianHoerst commented 5 months ago

Thanks for the issue, we are testing it in the following days.

FabianHoerst commented 5 months ago

Fixed!