Comments (9)
这个代码我用set1运行了,但是set1数据集下载下来不能直接使用,还需要做一点点处理,另外作者在论文中说得和实际在代码上运行有那么一点出入,我一直没训练到作者给我们的那种model的程度。
from deep_white_balance.
@xuhui1994 你是怎么做的?方便交流一下吗?
from deep_white_balance.
from deep_white_balance.
@TengFeiHan0 Yes, the code is only provided for training on Set1 with the augmented set. But, you can do some steps to make it works for other sets. You first need to copy all input/gt images into a single directory, such that for each image there are three ground-truth images with the same name but end with the following parts: 'G_AS.png', 'T_AS.png', 'S_AS.png' (for AWB, Tungsten WB, and Shade WB WB ground-truth, respectively). Other input images should also follow the same style; i.e., imagename_A_B.png, here A and B can be random characters or it can be the WB settings, and picture style, respectively (as we have in Set1).
Then, use fold=0 and the training should start without any problem.
from deep_white_balance.
@xuhui1994 if you have any questions, please let me know .. but in English please :).
from deep_white_balance.
@mahmoudnafifi I have downloaded your WB dataset, but I failed to unzip it correctly. The command I used is cat Set1_input_images.z* > wb.zip && unzip wb.zip
,after this I only got about 10000 images. I guess there have some problem when unzipping these files. If possible, would you mind providing a single link to the Set1 dataset rather than so many parts? Or how did you unzip your files under the Linux system?
from deep_white_balance.
I will find a way to upload them in a single zip file and back to you soon.
from deep_white_balance.
@mahmoudnafifi ok, thank you for help. I'm waiting for you. By the way, for set2 or cube+ dataset, there don't have the corresponding gt images end with ('G_AS.png', 'T_AS.png', 'S_AS.png')
, how did you generate it?
from deep_white_balance.
@TengFeiHan0 Yes, you need either to rename Cube+'s ground truth images to end with G_AS.png or maybe change the code.
One more thing, not all images in Set1 have ground-truth of T and S white-balance settings with the Adobe Standard (AS) picture style. So, you need to add a piece of code to select only those images that have ground-truth. Please, check this comment.
Update: I have uploaded Set 1 as a single zip file. Check it from here or here.
from deep_white_balance.
Related Issues (15)
- How to set dataset directory HOT 2
- Details about the results of WB manipulation(Table 2) HOT 2
- How to use wget download the dataset?
- About mapping_func HOT 2
- How to compute the MAE mean angular error in image HOT 1
- Automatic Image Rotation applied to output images HOT 1
- Unsigned Overflow in Python kernelP()
- When can you open source the code for exposure correction HOT 1
- hi,i want to use you model only use c++,could you give some suggestion? HOT 2
- why I get an error not found function or variable 'dlarray' HOT 1
- about using deep white balance in video files? HOT 1
- Example command in readme doesn't work HOT 1
- I need help of white balance case
- Evaluation metrics HOT 2
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 deep_white_balance.