import?os
from?PIL?import?Image

from?torch.utils.data?import?Dataset


class?KittiLoader（Dataset）:
????def?__init__（self?root_dir?mode?transform=None）:
????????left_dir?=?os.path.join（root_dir?‘image_02/data/‘）
????????self.left_paths?=?sorted（[os.path.join（left_dir?fname）?for?fname\
???????????????????????????in?os.listdir（left_dir）]）
????????if?mode?==?‘train‘:
????????????right_dir?=?os.path.join（root_dir?‘image_03/data/‘）
????????????self.right_paths?=?sorted（[os.path.join（right_dir?fname）?for?fname\
????????????????????????????????in?os.listdir（right_dir）]）
????????????assert?len（self.right_paths）?==?len（self.left_paths）
????????self.transform?=?transform
????????self.mode?=?mode


????def?__len__（self）:
????????return?len（self.left_paths）

????def?__getitem__（self?idx）:
????????left_image?=?Image.open（self.left_paths[idx]）
????????if?self.mode?==?‘train‘:
????????????right_image?=?Image.open（self.right_paths[idx]）
????????????sample?=?{‘left_image‘:?left_image?‘right_image‘:?right_image}

????????????if?self.transform:
????????????????sample?=?self.transform（sample）
????????????????return?sample
????????????else:
????????????????return?sample
????????else:
????????????if?self.transform:
????????????????left_image?=?self.transform（left_image）
????????????return?left_image

?屬性????????????大小?????日期????時間???名稱
-----------?---------??----------?-----??----
?????目錄???????????0??2018-10-28?13:15??MonoDepth-PyTorch-master\
?????文件??????????44??2018-10-28?13:15??MonoDepth-PyTorch-master\.gitignore
?????文件????????6406??2018-10-28?13:15??MonoDepth-PyTorch-master\Monodepth.ipynb
?????文件????????6160??2018-10-28?13:15??MonoDepth-PyTorch-master\README.md
?????文件????????1308??2018-10-28?13:15??MonoDepth-PyTorch-master\data_loader.py
?????文件????????7100??2018-10-28?13:15??MonoDepth-PyTorch-master\kitti_archives_to_download.txt
?????文件????????7208??2018-10-28?13:15??MonoDepth-PyTorch-master\loss.py
?????文件???????12692??2018-10-28?13:15??MonoDepth-PyTorch-master\main_monodepth_pytorch.py
?????文件???????15337??2018-10-28?13:15??MonoDepth-PyTorch-master\models_resnet.py
?????目錄???????????0??2018-10-28?13:15??MonoDepth-PyTorch-master\readme_images\
?????文件????23290849??2018-10-28?13:15??MonoDepth-PyTorch-master\readme_images\demo.gif
?????文件????????4932??2018-10-28?13:15??MonoDepth-PyTorch-master\transforms.py
?????文件????????2131??2018-10-28?13:15??MonoDepth-PyTorch-master\utils.py