Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

ImageNet pretrained Aligned Xception model #8

Open
PkuRainBow opened this issue May 24, 2019 · 11 comments
Open

ImageNet pretrained Aligned Xception model #8

PkuRainBow opened this issue May 24, 2019 · 11 comments

Comments

@PkuRainBow
Copy link

PkuRainBow commented May 24, 2019

Thanks for sharing the repo.

I notice that another repo mentions that there exist no ImageNet pre-trained Aligned Xception model, and the performance is worse than the ResNet-101.

In other words, I mean that the model below is not the Aligned Xception trained on ImageNet~

pretrain_dict = model_zoo.load_url('http://data.lip6.fr/cadene/pretrainedmodels/xception-b5690688.pth')

I am wondering whether you have trained the Aligned Xception model on the ImageNet. It would be great if you could share the models~

Besides, I am also wondering if the Imagnet pre-trained Aligned Xception model performs better than modified ResNet-101.

@Gaoyiminggithub
Copy link
Owner

Hi,
we have not trained the model on the ImageNet. Our model is trained on the COCO.
Maybe you can get the Xception pretrained on the ImageNet on here.

@PkuRainBow
Copy link
Author

@Gaoyiminggithub It seems that link is not trained on COCO.

I am wondering whether could you share the checkpoints of the Aligned Xception (instead of Xception) trained on COCO.

@Gaoyiminggithub
Copy link
Owner

@PkuRainBow
Hi, the link
https://github.com/tensorflow/models/blob/master/research/deeplab/g3doc/model_zoo.md
I provided above is the checkpoints of the Aligned Xception :) (you can see and find the imagenet and coco pretrained in it)
Maybe you confuse the link.

@PkuRainBow
Copy link
Author

PkuRainBow commented May 24, 2019

@Gaoyiminggithub Thanks for your quick reply, could you provide me the pytorch version or we can load the weights of the tensorflow models (in the Figure as below) directly?

image

@Gaoyiminggithub
Copy link
Owner

Gaoyiminggithub commented May 25, 2019

@PkuRainBow
I could not provide the pytorch version right now, because I delete it after I have trained the model on coco, but maybe I could convert that in next week.
You could convert the tensorflow models to the pytorch version by extracting the weight from the tensorflow models and then fill the pytorch model.weight with it.

@PkuRainBow
Copy link
Author

@Gaoyiminggithub Thanks for your help. I am wondering if you could share with me the mentioned COCO trained model.

@Gaoyiminggithub
Copy link
Owner

Gaoyiminggithub commented May 25, 2019

you can download the coco pretrained model in here.
The normalization way should follow here.

@PkuRainBow
Copy link
Author

@Gaoyiminggithub Thanks~

@PkuRainBow
Copy link
Author

PkuRainBow commented May 27, 2019

@Gaoyiminggithub Why do you comment the BN operation during the "Exit flow" as below?

        # Exit flow
        x = self.block20(x)
        x = self.conv3(x)
        # x = self.bn3(x)
        x = self.relu(x)

        x = self.conv4(x)
        # x = self.bn4(x)
        x = self.relu(x)

        x = self.conv5(x)
        # x = self.bn5(x)
        x = self.relu(x)

@Gaoyiminggithub
Copy link
Owner

@PkuRainBow because I put the BN operation in self.conv3/4/5.

@PkuRainBow
Copy link
Author

PkuRainBow commented May 27, 2019

@Gaoyiminggithub Got it~ Besides, it would be great if you could share with me the converted Pytorch model based on the official ImageNet based tensorflow models.

Besides, I want to check with you that the data normalization is simply to convert all the values to be in the range [-1, 1]?

Thanks a lot!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants