alinstein / Depth_estimation

Deep learning model to estimate the depth of image.
MIT License
113 stars 18 forks source link

Change '__getitem__' function in DepthDataset class #4

Open Taeyoung96 opened 3 years ago

Taeyoung96 commented 3 years ago

Hi! I'm using your great code.
I have a suggestion about __getitem__ function in DepthDataset class in UtilityTest.py.

I just execute the test_img.ipynb and I got a lot of depth prediction images.
In my case, It is important to get ordered data to match with input RGB images.
However, when I use these images, the order is different from the input images!

So, I just add sorted() function, to make the output in the same order as the input image list.

I think it is more helpful when you just add this function!

Thanks.