Name	Size	Mode	Actions
summary.cpython-36.pyc	22199	0644	editdlrm
writer.cpython-36.pyc	42668	0644	editdlrm
_caffe2_graph.cpython-36.pyc	24957	0644	editdlrm
_convert_np.cpython-36.pyc	1147	0644	editdlrm
_embedding.cpython-36.pyc	3200	0644	editdlrm
_onnx_graph.cpython-36.pyc	1887	0644	editdlrm
_proto_graph.cpython-36.pyc	1855	0644	editdlrm
_pytorch_graph.cpython-36.pyc	10052	0644	editdlrm
_utils.cpython-36.pyc	4309	0644	editdlrm
__init__.cpython-36.pyc	570	0644	editdlrm

3 �EgRt�@s�
dd
lZdd
l
Z
dd
lZdd
lZddlmZ
ddlmZ
ddl m Z
ddlmZ
ddlm Z
ddlmZ
ddlmZ
dd lmZ
dd lmZ
ddlmZ
ddlmZ
d dlmZ
d dlmZmZ
dd�ZdDdd�
ZdEdd�
Z dFdd�
Z!dd�Z"dGdd�
Z#dHdd �
Z$dId"d#�
Z%dJd%d&�
Z&dKd'd(�
Z'dLd)d*�
Z(dMd,d-�
Z)d.d/�Z*dNd1d2�
Z+d3d4�Z,d5d6�Z-dOd8d9�
Z.dPd:d;�
Z/dQdd?�Z1d@dA�Z2dRdBdC�
Z3d
S)S�N)
�Optional)
�range)
� struct_pb2)
�Summary)
�HistogramProto)
�SummaryMetadata)
�TensorProto)
�TensorShapeProto)
�TextPluginData)
�PrCurvePluginData)
� layout_pb2�
)
�make_np)�_prepare_video�convert_to_HWCc
Cs,t|t
j�s|j�n|}
|
jt
jkr(d
SdS)Nr ��)� isinstance�npZndarray�numpy�dtype�uint8)�tensorZ converted�r�K/usr/local/lib64/python3.6/site-packages/torch/utils/tensorboard/summary.py�_calc_scale_factors

r�black�c Cs�d
dlm
} m}
| j�}| j|�
}|
|||f\} }}}|j| |f| |f||f||f| |fg||d�
|r�|}|j|�
\}}tjd|�
}|j | ||d|f| ||fg|d�
|j | ||||f|||d�
|S)Nr
)� ImageDraw� ImageFont)�width�fillg�������?r)
r )r �font)�PILrrZload_defaultZDraw�line�getsizer�ceilZ rectangle�text)�imageZxminZyminZxmaxZymax�display_str�colorZ color_textZ thicknessrrr!Zdraw�left�right�topZbottomZtext_bottom� text_widthZtext_height�marginrrr�_draw_single_boxs"













r/cs�d
dl}d
dl
m}
d
dlm}m}m�m�
m}m }
d
dl m} m} m }m}m}
d
dlm}m}m}
t�t�s�tjd�

td�
�
t|
t�s�tjd�

td�
�
|p�i}t|t�s�td �
�
xR|j�D]F\�}��ks�t|t�s�t��fd d�|D�
�
r�tdj����
�
q�Wg}|�}�
x��j�D�
]�\�}|dk�
r8�
q t|t��
sPt|t��
r�||j�_ �|k�
r�t!j"d d�|�D�
d�
}nd}|j#|�|j$d�
|d��

�
q t||��r||j�_%�|k�
r�t!j"dd�|�D�
d�
}nd}|j#|�|j$d�
|d��

�
q t|t&��rd||j�_'�|k�rBt!j"dd�|�D�
d�
}nd}|j#|�|j$d�
|d��

�
q t||j(��r�t)|�
d
}||j�_ |j#|�|j$d�
d��

�
q t*d�
�
�
q W||| d�}t+t+j,| |j-�d�d�
}t.t.j$||d�g
d�
}��
fdd�|
j/�D�
}|||d�}||| d�}t+t+j,| |j-�d�d�
}t.t.j$||d�g
d�
}||j$d �
d!�
}||| d"�}t+t+j,| |j-�d�d�
}t.t.j$| |d�g
d�
}|||fS)#a2Outputs three `Summary` protocol buffers needed by hparams plugin. `Experiment` keeps the metadata of an experiment, such as the name of the hyperparameters and the name of the metrics. `SessionStartInfo` keeps key-value pairs of the hyperparameters `SessionEndInfo` describes status of the experiment e.g. STATUS_SUCCESS Args: hparam_dict: A dictionary that contains names of the hyperparameters and their values. metric_dict: A dictionary that contains names of the metrics and their values. hparam_domain_discrete: (Optional[Dict[str, List[Any]]]) A dictionary that contains names of the hyperparameters and all discrete values they can hold Returns: The `Summary` protobufs for Experiment, SessionStartInfo and SessionEndInfo r
N)
�string_types)� Experiment� HParamInfo� MetricInfo� MetricName�Status�DataType)�PLUGIN_NAME�PLUGIN_DATA_VERSION�EXPERIMENT_TAG�SESSION_START_INFO_TAG�SESSION_END_INFO_TAG)�HParamsPluginData�SessionEndInfo�SessionStartInfoz>parameter: hparam_dict should be a dictionary, nothing logged.z>parameter: metric_dict should be a dictionary, nothing logged.zIparameter: hparam_domain_discrete should be a dictionary, nothing logged.c
3s |]}
t|
t
��
�
�V
qdS)
N)r�type)�.0�
d)�hparam_dict�
krr� oszhparams..zWparameter: hparam_domain_discrete[{}] should be a list of same type as hparam_dict[{}].c
Ssg|]}
tj
|
d�
�qS)
)
�number_value)r�Value)r@rArrr� �s
zhparams..)
�valuesZDATA_TYPE_FLOAT64)�namer?�domain_discretec
Ssg|]}
tj
|
d�
�qS)
)
�string_value)rrF)r@rArrrrG�s
ZDATA_TYPE_STRINGc
Ssg|]}
tj
|
d�
�qS)
)
� bool_value)rrF)r@rArrrrG�s
ZDATA_TYPE_BOOL)rIr?z=value should be one of int, float, str, bool, or torch.Tensor)Zsession_start_info�version)�plugin_name�content)
�plugin_data)�tag�metadata)
�valuec
sg|]}
��
|
d�
d
�
�qS))
rQ)
rIr)r@rC)r3r4rrrG�s)Zhparam_infosZmetric_infos)Z experimentrMZSTATUS_SUCCESS)
�status)Zsession_end_inforM)0�torch�sixr0Z#tensorboard.plugins.hparams.api_pb2r1r2r3r4r5r6Z$tensorboard.plugins.hparams.metadatar7r8r9r:r;Z+tensorboard.plugins.hparams.plugin_data_pb2r<r=r>r�dict�logging�warning� TypeError�items�list�all�format�int�float�hparamsrErZ ListValue�appendrFrK�boolrLZTensorr� ValueErrorr� PluginData�SerializeToStringr�keys)rBZmetric_dictZhparam_domain_discreterUr0r1r2r5r6r7r8r9r:r;r<r=r>�
vZhpsZssirJrO�smdZmts�expZseir)r3r4rBrCrra3s�

























































raFcCs�t|
�
}
|
j
�jd
kstd�
�
t|
�
}
|rxt|
g
dd�}|rJt|
g
dd�}tjdd�
}t|d �
}ttj |||d �g
d�
Sttj ||
d�g
d�
Sd S)aOutputs a `Summary` protocol buffer containing a single scalar value. The generated Summary has a Tensor.proto containing the input Tensor. Args: name: A name for the generated node. Will also serve as the series name in TensorBoard. tensor: A real numeric Tensor containing a single value. collections: Optional list of graph collections keys. The new summary op is added to these collections. Defaults to `[GraphKeys.SUMMARIES]`. new_style: Whether to use new style (tensor field) or old style (simple_value field). New style could lead to faster data loading. Returns: A scalar `Tensor` of type `string`. Which contains a `Summary` protobuf. Raises: ValueError: If tensor has the wrong shape or type. r
zscalar should be 0D�DT_FLOAT)� float_valrZ DT_DOUBLE)Z double_valrZscalars)
rN)
rP)rQrrR)
rS)rQZsimple_valueN) r�squeeze�ndim�AssertionErrorr`rrrerrF)rI�scalar�collectionsZ new_styleZdouble_precisionrrPrirrrrp�s









rpc Cs,t|
||||||d
�}t
t
j||d�g
d�
S)aOutputs a `Summary` protocol buffer with a histogram. The generated [`Summary`](https://www.tensorflow.org/code/tensorflow/core/framework/summary.proto) has one summary value containing a histogram for `values`. Args: name: A name for the generated node. Will also serve as a series name in TensorBoard. min: A float or int min value max: A float or int max value num: Int number of values sum: Float or int sum of all values sum_squares: Float or int sum of squares for all values bucket_limits: A numeric `Tensor` with upper value per bucket bucket_counts: A numeric `Tensor` with number of values per bucket Returns: A scalar `Tensor` of type `string`. The serialized `Summary` protocol buffer. )�min�max�num�sum�sum_squares�bucket_limit�bucket)rQ�histo)
rS)rrrF) rIrrrsrtrurvZ bucket_limitsZ bucket_counts�histrrr� histogram_raw
s






r{cCs0t|
�
}
t
|
jt�
||�}ttj||d
�g
d�
S)adOutputs a `Summary` protocol buffer with a histogram. The generated [`Summary`](https://www.tensorflow.org/code/tensorflow/core/framework/summary.proto) has one summary value containing a histogram for `values`. This op reports an `InvalidArgument` error if any value is not finite. Args: name: A name for the generated node. Will also serve as a series name in TensorBoard. values: A real numeric `Tensor`. Any shape. Values to use to build the histogram. Returns: A scalar `Tensor` of type `string`. The serialized `Summary` protocol buffer. )rQry)
rS)r�make_histogram�astyper`rrF)rIrH�bins�max_binsrzrrr� histogram/
s

r�c Cs�
|jd
krt
d�
�
|jd�
}tj||
d�\}}t|�
}|dk r�||kr�||}||}|d
krztj|d
||gg
dd
d�}|jd|�jdd�
}tj|jdf
|j �}|dd|�|dd�<|d|d<|}tj tj|d
tjd ��
} tj | d
| ddgd d�\} }t| �
} t|�
d}~ | d
k�
r2|| d|�ntjd
g
|d|�g�
}|| |d�}|jd
k�
sr|jd
k�
rzt
d�
�
|j|�
}t|j�|j�t|�
|j�||j�|j�d �S)zDConvert values into a histogram proto using logic from histogram.cc.r
zThe input has no element.r )
r~NZconstant)Z pad_width�modeZconstant_values)
Zaxis)
rr+)
Zsidez1The histogram is empty, please file a bug report.)rrrsrtrurvrwrx�����r�r�r�r�r�r�r�)�sizerd�reshaperr��len�padru�emptyr�cumsumZgreater�int32Zsearchsortedr_Zconcatenate�dotrrrrs�tolist) rHr~r�countsZlimitsZnum_binsZsubsamplingZsubsampling_remainderZ new_limitsZ cum_counts�start�endZsum_sqrrrr|D
sB


















2








r|�NCHWcCsXt|
�
}
t
|
|�}
t|
�
}|
jtj�
}
|
|jtj�
}
t|
|d
�}ttj ||d�g
d�
S)aOutputs a `Summary` protocol buffer with images. The summary has up to `max_images` summary values containing images. The images are built from `tensor` which must be 3-D with shape `[height, width, channels]` and where `channels` can be: * 1: `tensor` is interpreted as Grayscale. * 3: `tensor` is interpreted as RGB. * 4: `tensor` is interpreted as RGBA. The `name` in the outputted Summary.Value protobufs is generated based on the name, with a suffix depending on the max_outputs setting: * If `max_outputs` is 1, the summary value tag is '*name*/image'. * If `max_outputs` is greater than 1, the summary value tags are generated sequentially as '*name*/image/0', '*name*/image/1', etc. Args: tag: A name for the generated node. Will also serve as a series name in TensorBoard. tensor: A 3-D `uint8` or `float32` `Tensor` of shape `[height, width, channels]` where `channels` is 1, 3, or 4. 'tensor' can either have values in [0, 1] (float32) or [0, 255] (uint8). The image() function will scale the image values to [0, 255] by applying a scale factor of either 1 (uint8) or 255 (float32). Returns: A scalar `Tensor` of type `string`. The serialized `Summary` protocol buffer. )
�rescale)rQr')
rS) rrrr}r�float32r� make_imagerrF)rQrr��dataformats�scale_factorr'rrrr'r
s




r'�CHWcCs\t|
�
}
t
|
|�}
t|�
}|
jtj�
t|
�
}
t|
jtj�
|||d
�}ttj ||d�g
d�
S)z0Outputs a `Summary` protocol buffer with images.)r��rois�labels)rQr')
rS) rrr}rr�rr�rrrF)rQZtensor_imageZtensor_boxesr�r�r�r'rrr�image_boxes�
s








r�c Csj|
jd
}t
|�
}xR|D]J}t||
|d
f|
|df|
|df|
|df|dkrTdn||dd�}qW|S)Nr
r r�ZRed)r(r))�shaperr/)Z disp_imageZboxesr�Z num_boxesZlist_gt�
irrr� draw_boxes�
s









r�cCs�d
dlm
}
|j\}}}t||
�
}t||
�
} |j|�
} |dk rPt| ||d�} | j| |f|j�} d
dl}|j �}| j |dd�
|j�} |j�
t j
|||| d�S)zg�?r r
)r~rr�r�r�r�r�) rr��floorr}Zfloat64r�r��maximumr�)r�r�r�r�Z_MINIMUM_COUNTZbucket_indicesZfloat_labelsZhistogram_rangeZ tp_buckets�
_Z fp_bucketsr�r�r�r�r�r�rrrr�\s.



















r�c Cs�d
dl}d
dl
m}
|j|�
}|j||
|||j||d�} td|jd�
j�t t j |jd
d�
t j |jdd�
t j |jdd�
gd �
d �}tj|j ||�|| d�} | S) aCreates a tensor summary with summary metadata. Args: name: Uniquely identifiable name of the summary op. Could be replaced by combination of name and type to make it unique even outside of this summary. display_name: Will be used as the display name in TensorBoard. Defaults to `name`. description: A longform readable description of the summary data. Markdown is supported. tensor: Tensor to display in summary. content_type: Type of content inside the Tensor. components: Bitmask representing present parts (vertices, colors, etc.) that belong to the summary. json_config: A string, JSON-serialized dictionary of ThreeJS classes configuration. Returns: Tensor summary with metadata. r
N)
rR)
�json_configrkr )
r�r)
r�)rrlr�)rQrrRr�)rU�tensorboard.plugins.meshrRZ as_tensorZcreate_summary_metadatar�rr�r�r r�rrFZget_instance_name)rI�display_name�descriptionrr�� componentsr�rUrRZtensor_metadataZtensor_summaryrrr�_get_tensor_summary{s,
 














r�c
Csd
}
|dk rtj
|dd�}
|
S)z6Parses and returns JSON string from python dictionary.z{}NT)
� sort_keys)�json�dumps)�config_dictr�rrr�_get_json_config�s


r�c Cs�d
dlm
}
d
dlm}
t|�
} g} |
|jf||jf||jfg}dd�|D�
}|jdd�|D�
�
}x*|D]"\} }| j t |||| ||| ��

qhWt| d�
S)aHOutputs a merged `Summary` protocol buffer with a mesh/point cloud. Args: tag: A name for this summary operation. vertices: Tensor of shape `[dim_1, ..., dim_n, 3]` representing the 3D coordinates of vertices. faces: Tensor of shape `[dim_1, ..., dim_n, 3]` containing indices of vertices within each triangle. colors: Tensor of shape `[dim_1, ..., dim_n, 3]` containing colors for each vertex. display_name: If set, will be used as the display name in TensorBoard. Defaults to `name`. description: A longform readable description of the summary data. Markdown is supported. config_dict: Dictionary with ThreeJS classes names and configuration. Returns: Merged summary for mesh/point cloud representation. r
)
�MeshPluginData)
rRc
Ssg|]}
|
dd
k r|
�qS)r
Nr)r@rrrrrG�szmesh..c
Ssg|]\}
}|�qSrr)r@rr�rrrrG�s)
rS)Z(tensorboard.plugins.mesh.plugin_data_pb2r�r�rRr�ZVERTEXZFACEZCOLORZget_components_bitmaskrbr�r)rQZvertices�colorsZfacesr�r�r�r�rRr�Z summariesZtensorsr�rr�rrr�mesh�s







r�)rrr)NNN)NFF)
N)
N)r r�)r r�N)
N)r NN)
r�)
r�)r�N)r�N)NN)NN)4r�rXrrr��typingrZ six.movesrZgoogle.protobufrZ$tensorboard.compat.proto.summary_pb2rrrZ#tensorboard.compat.proto.tensor_pb2rZ)tensorboard.compat.proto.tensor_shape_pb2r Z(tensorboard.plugins.text.plugin_data_pb2r Z,tensorboard.plugins.pr_curve.plugin_data_pb2rZ!tensorboard.plugins.custom_scalarrZ_convert_npr�_utilsrrrr/rarpr{r�r|r'r�r�r�r�r�r�r�r&r�r�r�r�r�r�rrrr�
sN













  7 (  . #    '    4