Comments (6)
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.
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.
@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.
@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.
from gcc-sfcn.
That's exactly what I want to see, very interesting results. Thanks for posting it here!
from gcc-sfcn.
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
Thanks for your attention!
from gcc-sfcn.
Related Issues (20)
- Cannot reproduce the results for SHTA and UCF-QNRF dataset with cyclegan translated images
- How can I use the GCC dataset in my training process? HOT 1
- How to Accelerate Operation? HOT 1
- Idea: combine supervised and domain adaptation methods HOT 4
- RuntimeError: CUDNN_STATUS_INTERNAL_ERROR HOT 2
- When I want to train with my own data set, I get the following error
- IndexError happend in training HOT 1
- The segmentation file generated by the code of mcnn?
- Test my own data HOT 1
- .csv file doesn't exit when I run the python text.py HOT 1
- flops and params of the model ?
- Is this okay that you validate on test set? HOT 1
- Why divide predicted result by 100 in test.py? HOT 1
- where “”resnet101-5d3b4d8f.pth“”?? HOT 2
- What is the configuration of your computer for experiment? HOT 1
- SSIM loss
- What is the speed of the network ?
- Final model for the ShanghaiTech B dataset
- Crowd Counting problems
- where is the final model?can i test the model on my own picture directly? HOT 3
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
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.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from gcc-sfcn.