Lite-HRNet icon indicating copy to clipboard operation
Lite-HRNet copied to clipboard

Can the model be modified for segmentation task?

Open BJTUJia opened this issue 4 years ago • 3 comments

Hi there, thanks for this nice work! I am trying to modify the Lite-HRNet model on my segmentation task with my own data, by replacing the head with a seg head to predict segmentation maps, as shown in the picture. image

However, the results are pretty bad. I think the possible reasons are 1:) the model cannot be directly utilized for segmentation?If so, can you release the model for seg? 2:) the training is unstable due to the lack of pretrained weights? In this case, is it possible to release the ImgaNet pre-trained weights for Lite-HRNet, like the ones in https://github.com/HRNet/HRNet-Semantic-Segmentation?

Any kind of help or assistance is highly appreciated! Thanks in advance!

BJTUJia avatar Apr 30 '21 03:04 BJTUJia

Similar to your method, it did not work well on my task, and the training was very slow.

huixiancheng avatar May 03 '21 13:05 huixiancheng

Hi, kind of on the same boat. Segmentation results are much worse than what is reported despite the experiment setting being the same. Hope the authors release their segmentation code.

nsapkota417 avatar Jan 18 '22 03:01 nsapkota417

hope for a feasible solution

minushuang avatar Sep 01 '22 12:09 minushuang