Zxinze 9776de95dd | 1 year ago | |
---|---|---|
data | 1 year ago | |
dataset | 1 year ago | |
models | 1 year ago | |
options | 1 year ago | |
results | 1 year ago | |
util | 1 year ago | |
README.md | 1 year ago | |
loss.py | 1 year ago | |
metrics.py | 1 year ago | |
test.py | 1 year ago | |
train.py | 1 year ago | |
util.py | 1 year ago |
You can run our training code in gvien examples of dataset folder by following instruction.
python train.py --dataroot=./dataset --batch_size=1
Models are saved to ./checkpoints/
.
See opt
in files(base_options.py and train_options.py) for additional training options.
You can test our pretrained model in gvien examples of dataset folder by following instruction.
python test.py --dataroot=./dataset --epoch=200 --data_type=short
'--epoch=200' means loading the 200th epoch saved model.
'--data_type=short or long', 'short' means testing imges of short exposure time, while 'long' means testing imges of long exposure time (saturation images).
The pretrained model can be downloaded at Baidu Netdisk, password: tr2n
.
Testing results are saved in ./result/
, and each result inculdes 8 images, IA, IM, IG, Icomp, IGT, ISAT, M(IG), M(IGT), Where the mean of these tokens are in our paper.
See opt
in files (base_options.py and test_options.py) for additional testing options.
Several samples of our dataset are in ./dataset
. Each sample incules 3 images, ISAT, IGT, IM. The size of training image is 350 x 350, but they are randomly crop to 256 x 256 in training process. Testing image is 256 x 256.
Code borrows heavily from pytorch-CycleGAN-and-pix2pix and partialconv.
Dear OpenI User
Thank you for your continuous support to the Openl Qizhi Community AI Collaboration Platform. In order to protect your usage rights and ensure network security, we updated the Openl Qizhi Community AI Collaboration Platform Usage Agreement in January 2024. The updated agreement specifies that users are prohibited from using intranet penetration tools. After you click "Agree and continue", you can continue to use our services. Thank you for your cooperation and understanding.
For more agreement content, please refer to the《Openl Qizhi Community AI Collaboration Platform Usage Agreement》