Custom dataset problem

[fanshixiong]

Hi, thank you very much for your work. Excuse me, I put the pictures in the nyu dataset directory, and modified the nyu_test file, but the program does not run normally, but the nyu dataset can run normally, is it because of the resolution or what?
My data resolution is 3840 * 2160, what needs to be modified?

[zhyever]

I was wondering if this process is out of memory? And it would be killed by the system.