o 2h@sLdZddlZddlmZGdddejZeZGdddejZeZ dS)zCTensorBoard encoder helper module. Encoder depends on TensorFlow. N) op_evaluatorc0eZdZdZfddZddZddZZS)_TensorFlowPngEncoderabEncode an image to PNG. This function is thread-safe, and has high performance when run in parallel. See `encode_png_benchmark.py` for details. Arguments: image: A numpy array of shape `[height, width, channels]`, where `channels` is 1, 3, or 4, and of dtype uint8. Returns: A bytestring with PNG-encoded data. cstd|_d|_dSN)super__init___image_placeholder _encode_opself __class__S/var/www/html/chatgem/venv/lib/python3.10/site-packages/tensorboard/util/encoder.pyr(s  z_TensorFlowPngEncoder.__init__cCs8ddlmm}|j|jdd|_|j|j|_dS)Nrimage_to_encodedtypename) tensorflow.compat.v1compatv1 placeholderuint8rimage encode_pngr r tfrrrinitialize_graph-s z&_TensorFlowPngEncoder.initialize_graphcCsFt|tjs td||jtjkrtd|j|jj|j|idS)Nz!'image' must be a numpy array: %rz&'image' dtype must be uint8, but is %r feed_dict) isinstancenpndarray ValueErrorrrr evalr)r rrrrrun6s   z_TensorFlowPngEncoder.run__name__ __module__ __qualname____doc__rrr% __classcell__rrr rrs   rcr)_TensorFlowWavEncodera!Encode an audio clip to WAV. This function is thread-safe and exhibits good parallel performance. Arguments: audio: A numpy array of shape `[samples, channels]`. samples_per_second: A positive `int`, in Hz. Returns: A bytestring with WAV-encoded data. cs td|_d|_d|_dSr)rr_audio_placeholder_samples_per_second_placeholderr r r rrrPs  z_TensorFlowWavEncoder.__init__cCsPddlmm}|j|jdd|_|j|jdd|_|jj |j|jd|_ dS)Nrrrsamples_per_second) sample_rate) rrrrfloat32r-int32r.audio encode_wavr rrrrrVs z&_TensorFlowWavEncoder.initialize_graphcCsLt|tjs td|t|tstd||j||j|i}|jj|dS)Nz!'audio' must be a numpy array: %rz''samples_per_second' must be an int: %rr) r r!r"r#intr-r.r r$)r r3r/rrrrr%es   z_TensorFlowWavEncoder.runr&rrr rr,Cs  r,) r*numpyr!tensorboard.utilrPersistentOpEvaluatorrrr,r4rrrrs & 2