search

zym1119 / DeepLabv3_MobileNetv2_PyTorch

A PyTorch Implementation of MobileNetv2+DeepLabv3
151 stars 31 forks source link

Hi~Have you used the moblie net which is pretrained on the ImageNet? #5

Open irfanICMLL opened 6 years ago

irfanICMLL commented 6 years ago

Hello~ I am working on training the mobileNetV2+aspp these days. I found out if I only train the model on the cityscapes, the mIoU is very low. Do you use the mobile net which is pretrained on the ImageNet? Can you share the weigh of the mobile net V2 in pytorch?

zym1119 commented 6 years ago

Sorry I dont have the imagenet pre-trained weight, cause I dont have ImageNet in my computer. I got a little busy recently :-) , I will try to get others pre-trained weight in my spare time

zym1119 commented 6 years ago

i upload a ImageNet pre-trained weight, you can try it :)

irfanICMLL commented 6 years ago

So you mean you just train from the scratch, and get the IoU of 70?

irfanICMLL commented 6 years ago

I loaded in your ImageNet pre-trained weight, but most of the weights are zeros or ones. It seemed the parameters had not been trained.

zym1119 commented 6 years ago

This ImageNet model is downloaded from another repo of MobileNetv2 on github with most stars, i have no time checking on this carefully cause im busy working on my company project, maybe something is wrong when transfering weight, sorry to waste your time on my code. Maybe after going back to school can i have time to update this. I got only 60+ mIoU on cityscapes on my off-line code, but have no time to rearrange and update.