TIO-IKIM / CellViT

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

Error report for cellvit_cpp_net.py #49

Closed won5830 closed 1 month ago

won5830 commented 1 month 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 1 month ago

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

FabianHoerst commented 1 month ago

Fixed!