Comments (11)
Hey, I am here to help. Actually 34.62-34.65 is around the performance of EDSR. So I wonder have you checked that you are calling WDSR model? Can you print the model size (which should be the same as our report)?
We do have end-to-end training code on tensorflow. It could be a good reference for your implementation.
from wdsr_ntire2018.
I am sure that WDSR is set. the model and parameters will be printed for each train. I have tried to reproduce several times, but each time the results obtained is 36.65dB. I don't know if this is related to OS (mine is win10). In addition, can you provide the args setting in option.py that you get 36.77dB? THX
from wdsr_ntire2018.
@ChaofWang Hey, we already provide the args settings that we changed based on EDSR. In your case on your OS, what's the performance of EDSR?
from wdsr_ntire2018.
Yeah,I tried EDSR several times, and the results were 34.55-34.57 dB, which also are lower than README report. It may be related to OS. I will reinstall my OS and continue to try on Ubuntu 16.04.
from wdsr_ntire2018.
@ChaofWang Since the improvement of WDSR is consistent compared with EDSR, there may be some other issues with your code base. Please let us know if you have further question.
ps: my personal feeling is that OS may not be the reason. Maybe you can check your dataset, python version, opencv version and other potential bugs. You can also try our tensorflow version re-implementation on your windows which provides an end-to-end training-validation pipeline.
from wdsr_ntire2018.
@JiahuiYu I reinstall my OS and retrain both WDSR baselinex2 and EDSR baselinex2 in your table1. Actually,you are right,OS is not the reason. But now, I can reproduce the 34.61dB of EDSR and the 34.68dB of WDSR. The result of WDSR is the same as keras version, both EDSR and WDSR without self-ensemble . But I found after WDSR used self-ensemble, the result just is 34.772dB. So I think that your description of Table 1 about WDSR without self-ensemble may be a mistake, or can you provide the pre-training model for pytorch? These results are reproduced in the framework of EDSR-Pytorch, and both in WIN10 and ubuntu16.04 used python3.5, CUDA9.0, cudnn7.1. Another interesting thing, in the same CUDA environment, including the same disk, ubuntu16.04 training is twice as fast as win10.
from wdsr_ntire2018.
We do have tensorflow pretrained model available. You can probably convert it to pytorch using other tools. We are not using self-ensemble in table 1.
By the way, if you have time, can you try to reproduce table 2 (or maybe some settings in table 2), and let me know the results?
from wdsr_ntire2018.
Model | Number of Residual Blocks | Parameters | PSNR |
---|---|---|---|
EDSR | 3 | 409755 | 33.997 dB |
WDSRA | 3 | 227476 | 34.107 dB |
EDSR | 5 | 557467 | 34.218 dB |
WDSRA | 5 | 375572 | 34.325 dB |
EDSR | 8 | 779035 | 34.434 dB |
WDSRA | 8 | 597716 | 34.504 dB |
@JiahuiYu,This is my implementation result of Table 2. All results are lower than yours, but it also proves that WDSR is better. In addition, I don't care about the difference between tensorflow and pytorch, nor whether it can achieve this result on tensorflow. I just want to know why the results of your pytorch implementation are higher than mine. Is there any tricks in training here? Or, can you tell me your training details on pytorch? This has bothered me for a long time. THX
from wdsr_ntire2018.
@ChaofWang Hey, thanks for verifying the improvements. In fact, I also want to help you achieve better results. But indeed I don't have any other secret tricks. Did you implement SR framework by yourself, or based on EDSR-Pytorch?
from wdsr_ntire2018.
@JiahuiYu yeah,Those result are all based on EDSR-Pytorch. Are your results based on EDSR-pytorch?
from wdsr_ntire2018.
@ChaofWang Yes, it is kind of strange that a performance gap exists.
from wdsr_ntire2018.
Related Issues (20)
- Testing the models HOT 4
- Testing on "wild LR" = track 3 NTIRE2018 HOT 1
- How to reduce inference time HOT 1
- I didn't get desired training result HOT 3
- An error when training with my own dataset with scale 3 or 4 HOT 1
- How to achieve the results in the paper? HOT 1
- preprocess for training data HOT 1
- About the linear low-rank convolution HOT 1
- loss suddenly turned into nan HOT 1
- Some conceptual questions HOT 3
- Where weight normalization is used? HOT 1
- I got a result image of all gray with no contents HOT 3
- What does this do? x = (x - self.rgb_mean.cuda()*255)/127.5 HOT 1
- Ablation study for WDSR-B HOT 1
- There is not the second act layer in WDSR-B between the 3x3 Conv and the second 1x1 Conv, why? HOT 1
- In WDSR-B, why not do the 3x3 Conv without the second 1x1 Conv? HOT 1
- in forward of wdsr.py, where dose 127.5 come from? HOT 4
- Pytorch pretrained model for WDSR HOT 1
- Weight normalization during inference. HOT 1
- Custom dataset training
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 wdsr_ntire2018.