python test.py -weights -dataset -in_nc
weights:Weight path
dataset:dataset path
in_nc:input channel
example:
python test.py -dataset AIM2020_ISP_test_raw/ -weights ./save_model/raw2rgb_15.pth -in_nc 4
You can download the model from the link below
(3 channel input) : https://drive.google.com/file/d/1Q0fK55X1zfpp4VAD80Y_3GC38i6IlCQb/view?usp=sharing
(4 channel input): https://drive.google.com/file/d/1ag2l_xpquIM2dUTRyt1jSYvvsYcZhwlA/view?usp=sharing