Giter VIP home page Giter VIP logo

Comments (6)

gjy3035 avatar gjy3035 commented on May 24, 2024

vgg network? you mean sfcn based on vgg or pure vgg? we have conducted two baselines with vgg backbone. unfortunately, the trained models are saved in my server. when i return lab, i can share the pretrained weights and the model definitions with you.

from gcc-sfcn.

xialeiliu avatar xialeiliu commented on May 24, 2024

Pure vgg, I want to compare training from ImageNet-pretrained and GCC-pretrained VGG-16.
Thanks a lot!
By the way, have you observed that GCC-pretrained network is better than ImageNet-pretrained?
I guess you compare these two in your paper.

from gcc-sfcn.

Elin24 avatar Elin24 commented on May 24, 2024

@xialeiliu I am a co-author of this paper. The code you want is as follows

VGG :

class VGG(nn.Module):
    def __init__(self, pretrained=True):
        super(VGG, self).__init__()
        vgg = models.vgg16()
        if pretrained:
            vgg.load_state_dict(torch.load(model_path))
        features = list(vgg.features.children())
        self.features4 = nn.Sequential(*features[0:23])

        self.de_pred = nn.Sequential(
            Conv2d(512, 128, 1, same_padding=True, NL='relu'),
            Conv2d(128, 1, 1, same_padding=True, NL='relu')
        )

    def forward(self, x):
        x = self.features4(x)       
        x = self.de_pred(x)
        x = F.upsample(x,scale_factor=8)
        return x

VGG decoder:

class VGG_decoder(nn.Module):
    def __init__(self, pretrained=True):
        super(VGG_decoder, self).__init__()
        vgg = models.vgg16()
        if pretrained:
            vgg.load_state_dict(torch.load(model_path))
        features = list(vgg.features.children())
        self.features4 = nn.Sequential(*features[0:23])

        self.de_pred = nn.Sequential(
            Conv2d( 512, 128, 3, same_padding=True, NL='relu'),
            nn.ConvTranspose2d(128,64,4,stride=2,padding=1,output_padding=0,bias=True),
            nn.ReLU(),
            nn.ConvTranspose2d(64,32,4,stride=2,padding=1,output_padding=0,bias=True),
            nn.ReLU(),
            nn.ConvTranspose2d(32,16,4,stride=2,padding=1,output_padding=0,bias=True),
            nn.ReLU(),
            Conv2d(16, 1, 1, same_padding=True, NL='relu')
        )

    def forward(self, x):
        x = self.features4(x)       
        x = self.de_pred(x)
        return x

@gjy3035 will provide the pretrained vgg model parameters later, if you need.

from gcc-sfcn.

gjy3035 avatar gjy3035 commented on May 24, 2024

@xialeiliu in our paper, we have compared the results using different installation. due to other stuff, our paper can not be polished. we will upload the paper in 5 days.
41a005c5-d589-470f-9f40-756c979054a2

from gcc-sfcn.

xialeiliu avatar xialeiliu commented on May 24, 2024

That's exactly what I want to see, very interesting results. Thanks for posting it here!

from gcc-sfcn.

gjy3035 avatar gjy3035 commented on May 24, 2024

We have upload the download link for VGG and VGG_decoder, which are trained under the C^3 Framework (the performance is better this repo).
https://mailnwpueducn-my.sharepoint.com/:u:/g/personal/gjy3035_mail_nwpu_edu_cn/EVdYCMGL5WRKp-QRSXs6KlsBGinp0XA3KR4No9cC0OuBcw?e=nehYUH

https://mailnwpueducn-my.sharepoint.com/:u:/g/personal/gjy3035_mail_nwpu_edu_cn/EZNmWR2Si25Auz3yGhrjXYUBgcGVY84yCslFS_pHYTqAtw?e=zYtXb0

Thanks for your attention!

from gcc-sfcn.

Related Issues (20)

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.