orhan-t / SM-CNN

7 stars 1 forks source link

Can you tell me how much data I need to prepare for training the model to achieve the score in the paper? #2

Open KikkiZ opened 9 months ago

KikkiZ commented 9 months ago

Can you tell me how much data I need to prepare for training the model to achieve the score in the paper? I tried using the WDC dataset, code automatically processed the data and generated 1024 samples. When training with these samples, the MPSNR of the model was about 28 or so. Aftertraining for 50 epochs, when testing with 200 x 200 x 191 size data, the test result is far from the results in the paper, it's only around 21.8. I wonder if it's a problem with too few samples used during training?