STDC-Seg
STDC-Seg copied to clipboard
Issues about loading pretrained model.
The implement uses init_weight to load the pretrained model. However, you use nn.Sequential to repackage the nn.Modules. So, the code has two sets of parmaters, and you only load weight to the first sets. That is, the second set (for segmentation) are trained from scratch.