o
    Ni9
                     @   s\   d Z ddlmZ ddlmZ ddlmZ ddlZddlmZ dZ	dZ
G dd	 d	ejjZdS )
zFlowers dataset.
    )absolute_import)division)print_functionNz@ONLINE {tfflowers,
author = "The TensorFlow Team",
title = "Flowers",
month = "jan",
year = "2019",
url = "http://download.tensorflow.org/example_images/flower_photos.tgz" }
z?http://download.tensorflow.org/example_images/flower_photos.tgzc                   @   s4   e Zd ZdZejdZdd Zdd Z	dd Z
d	S )
	TFFlowerszFlowers dataset.z3.0.1c              	   C   s:   t jj| dt jt j t jjg dddddtdS )Nz A large set of images of flowers)	dandeliondaisytulips
sunflowersroses)namesimagelabelz5https://www.tensorflow.org/tutorials/load_data/images)builderdescriptionfeaturessupervised_keyshomepagecitation)tfdscoreDatasetInfor   FeaturesDictImage
ClassLabel	_CITATION)self r   d/home/ubuntu/.local/lib/python3.10/site-packages/tensorflow_datasets/image_classification/flowers.py_info-   s   zTFFlowers._infoc                 C   s*   | t}tjjtjjd||idgS )Nimages_dir_path)name
gen_kwargs)download_URLr   r   SplitGeneratorSplitTRAINiter_archive)r   
dl_managerpathr   r   r   _split_generators;   s   

zTFFlowers._split_generatorsc                 c   sZ    |D ]'\}}| dr*tj|\}}tj|}|| d}d||f |fV  qdS )zGenerate flower images and labels given the image directory path.

    Args:
      images_dir_path: path to the directory where the images are stored.

    Yields:
      The image path and its corresponding label.
    z.jpgr   z%s/%sN)endswithosr*   splitbasenamelower)r   r    fnamefobj	image_dir
image_filedrecordr   r   r   _generate_examplesG   s   	
zTFFlowers._generate_examplesN)__name__
__module____qualname____doc__r   r   VersionVERSIONr   r+   r7   r   r   r   r   r   (   s    r   )r;   
__future__r   r   r   r-   tensorflow_datasets.public_api
public_apir   r   r$   r   GeneratorBasedBuilderr   r   r   r   r   <module>   s   	