o
    Ni                     @   sz   d Z ddlmZ ddlmZ ddlmZ ddlZddlZddlm	  m
Z ddlmZ dZdZdZG d	d
 d
ejjZdS )NYU Depth V2 Dataset.    )absolute_import)division)print_functionNa+  @inproceedings{Silberman:ECCV12,
  author    = {Nathan Silberman, Derek Hoiem, Pushmeet Kohli and Rob Fergus},
  title     = {Indoor Segmentation and Support Inference from RGBD Images},
  booktitle = {ECCV},
  year      = {2012}
}
@inproceedings{icra_2019_fastdepth,
  author    = {Wofk, Diana and Ma, Fangchang and Yang, Tien-Ju and Karaman, Sertac and Sze, Vivienne},
  title     = {FastDepth: Fast Monocular Depth Estimation on Embedded Systems},
  booktitle = {IEEE International Conference on Robotics and Automation (ICRA)},
  year      = {2019}
}
zThe NYU-Depth V2 data set is comprised of video sequences from a variety of
indoor scenes as recorded by both the RGB and Depth cameras from the
Microsoft Kinect.
z=http://datasets.lids.mit.edu/fastdepth/data/nyudepthv2.tar.gzc                   @   s4   e Zd ZdZejdZdd Zdd Z	dd Z
d	S )

NyuDepthV2r   z0.0.1c              
   C   s>   t jj| tt jt jjddt jjdtj	ddddt
dS )N)       )shape)r   r   )r
   dtypeimagedepthz8https://cs.nyu.edu/~silberman/datasets/nyu_depth_v2.html)builderdescriptionfeaturessupervised_keyshomepagecitation)tfdscoreDatasetInfo_DESCRIPTIONr   FeaturesDictImageTensortffloat16	_CITATION)self r    Z/home/ubuntu/.local/lib/python3.10/site-packages/tensorflow_datasets/image/nyu_depth_v2.py_info=   s   zNyuDepthV2._infoc              	   C   sR   | t}tjjtjjdtj	|ddidtjjtjj
dtj	|ddidgS )zReturns SplitGenerators.root_dir
nyudepthv2train)name
gen_kwargsval)download_and_extract_URLr   r   SplitGeneratorSplitTRAINospathjoin
VALIDATION)r   
dl_manager	base_pathr    r    r!   _split_generatorsJ   s   
zNyuDepthV2._split_generatorsc              
   c   s    t jjj}tjj|D ]H}tjjtj	
||D ]9}|tj	
|||d#}|d | t|d d|d dd ddfV  W d   n1 sOw   Y  qqdS )	zYields examples.r_rgb)      r   r   Nr   r   )r   r   lazy_importsh5pyr   iogfilelistdirr.   r/   r0   Filenp	transposeastype)r   r#   r;   	directory	file_namefr    r    r!   _generate_examples^   s   

zNyuDepthV2._generate_examplesN)__name__
__module____qualname____doc__r   r   VersionVERSIONr"   r4   rF   r    r    r    r!   r   8   s    r   )rJ   
__future__r   r   r   r.   numpyr@   tensorflow.compat.v2compatv2r   tensorflow_datasets.public_api
public_apir   r   r   r*   r   GeneratorBasedBuilderr   r    r    r    r!   <module>   s   