1Vh#dZddlZddlZddlmZddlmZddlmZddl m Z ddl m Z ddl mZdd lmZd d d d ddZdZdZdZej*j-eGddej.Zy)zThe TensorBoard Images plugin.N)wrappers)errors) plugin_util) http_util)provider) base_plugin)metadataz image/bmpz image/gifz image/jpegz image/pngz image/svg+xml)bmpgifjpegpngsvgzapplication/octet-stream cL~|jds|jdryy)Ns&?c eZdZdZej ZdZdZdZ dZ dZ e jjdZdZd Zd Zd Zd Ze jjd Ze jjdZy) ImagesPluginzImages Plugin for TensorBoard.c|jxsij|jt|_|j |_tjdd|_ y)z|Instantiates ImagesPlugin via TensorBoard core. Args: context: A base_plugin.TBContext instance. imager) data_kindlatest_known_versionN) sampling_hintsget plugin_name_DEFAULT_DOWNSAMPLING_downsample_to data_provider_data_providerr_MetadataVersionChecker_version_checker)selfcontexts r__init__zImagesPlugin.__init__;sW '55;@@   3 &33 + C C!"! rcJ|j|j|jdS)N)z/imagesz/individualImagez/tags)_serve_image_metadata_serve_individual_image _serve_tagsr&s rget_plugin_appszImagesPlugin.get_plugin_appsJs'11 $ < <%%  rcy)NFr-s r is_activezImagesPlugin.is_activeQsrc.tjdS)Nztf-image-dashboard) element_name)rFrontendMetadatar-s rfrontend_metadatazImagesPlugin.frontend_metadataTs++9MNNrc|jj||tj}|Dcic]}|i}}|j D]\}}|j D]\}}tj |j } |jj| j||sMtj|j} |j| |jdz d|||<|Scc}w)N) experiment_idr) displayName descriptionsamples)r#list_blob_sequencesr PLUGIN_NAMEitemsparse_plugin_metadataplugin_contentr%okversionrmarkdown_to_safe_htmlr: display_name max_length) r&ctx experimentmappingrunresulttag_to_contenttag metadatummdr:s r _index_implzImagesPlugin._index_implWs%%99 $ ,,:  &--c#r'--#*==?  C"0"6"6"8 Y33I4L4LM,,// CE)??)) $-#9#9#.(33a7$s C    .s C8ctj|j}tj|j}|jj d}|jj d}t |jj dd} |j|||||}tj||d S#t$rtj|dddcYSwxYw) aGiven a tag and list of runs, serve a list of metadata for images. Note that the images themselves are not sent; instead, we respond with URLs to the images. The frontend should treat these URLs as opaque and should not try to parse information about them or generate them itself, as the format may change. Args: request: A werkzeug.wrappers.Request object. Returns: A werkzeug.Response application. rLrIsamplerzInvalid run or tag text/plaincodeapplication/json) rr'environr7argsrint_image_response_for_runKeyErrorrRespond)r&requestrFrGrLrIrQresponses rr*z"ImagesPlugin._serve_image_metadatams!!'//2 ..w? llu%llu%W\\%%h23 33Zc6H  (4FGG  $$-|#  sC"C,+C,c |jj||tj|jt j |g|g}|j|ij|d}|tjd|d||Dcgc]V}t|jdz |kDr9|j|j|j|j|dzdXc}Scc}w)aBuilds a JSON-serializable object with information about images. Args: run: The name of the run. tag: The name of the tag the images all belong to. sample: The zero-indexed sample of the image for which to retrieve information. For instance, setting `sample` to `2` will fetch information about only the third image of each batch. Steps with fewer than three images will be omitted from the results. Returns: A list of dictionaries containing the wall time, step, and URL for each image. Raises: KeyError, NotFoundError: If no image data exists for the given parameters. )runstags)r7r downsamplerun_tag_filterNzNo image data for run=z, tag=r8) wall_timestepquery)r#read_blob_sequencesr r=r!r RunTagFilterrr NotFoundErrorlenvaluesrdre_data_provider_query) r&rFrGrIrLrQ all_imagesimagesdatums rrZz$ImagesPlugin._image_response_for_runs&((<< $ ,,**#00se3%H = R(,,S$7 >&&693?   5<< 1$v- #__ 225<< 3KL    sAC1cv|Dcgc])}t|jjdz |kDr|+c}Scc}w)Nr8)rj tensor_proto string_val)r& tensor_eventsrQ tensor_events r_filter_by_samplezImagesPlugin._filter_by_samplesE!. L--889A=    s.6cNtjj||||d}|S)aBuilds a URL for accessing the specified image. This should be kept in sync with _serve_image_metadata. Note that the URL is *not* guaranteed to always return the same image, since images may be unloaded from the reservoir as new images come in. Args: run: The name of the run. tag: The tag. sample: The relevant sample index, zero-indexed. See documentation on `_image_response_for_run` for more details. index: The index of the image. Negative values are OK. Returns: A string representation of a URL that will load the index-th sampled image in the given run with the given tag. )rIrLrQindex)urllibparse urlencode)r&rIrLrQrw query_strings r_query_for_individual_imagez(ImagesPlugin._query_for_individual_images1$||--    rcXtjjd|jiS)Nblob_key)rxryrzr~)r&blob_references rrlz!ImagesPlugin._data_provider_querys"||%%z>3J3J&KLLrc<|jj||S)zReturns the actual image bytes for a given image. Args: blob_key: As returned by a previous `read_blob_sequences` call. Returns: A bytestring of the raw image bytes. )r~)r# read_blob)r&rFr~s r"_get_generic_data_individual_imagez/ImagesPlugin._get_generic_data_individual_images "",,S8,DDrcn tj|j}|jd}|j ||}tjd|}tj|t}tj|||S#t t f$rtj|dddcYSwxYw)zServes an individual image.r~z"Invalid run, tag, index, or samplerRrSrTN)rr'rWrXrr[ IndexErrorrr\imghdrwhat_IMGHDR_TO_MIMETYPEr_DEFAULT_IMAGE_MIMETYPE)r&r]rFr~r image_type content_types rr+z$ImagesPlugin._serve_individual_images %%goo6C||J/H::3ID[[t, *.. /   $ ==*% $$4   sAB (B43B4ctj|j}tj|j}|j ||}t j ||dS)NrV)rr'rWr7rOrr\)r&r]rFrGrws rr,zImagesPlugin._serve_tagssR!!'//2 ..w?   j1  %1CDDrN)__name__ __module__ __qualname____doc__r r=rr(r.r1r5rOrRequest applicationr*rZrur|rlrr+r,r0rrrr6s(&&K   O,!!H"H:' R 8M E!!>">&!!E"Err)rr urllib.parserxwerkzeugr tensorboardrrtensorboard.backendrtensorboard.datartensorboard.pluginsrtensorboard.plugins.imager rrr rtestsappendTBPluginrr0rrrs}%#)%+.      5 JKE;''KEr