U <cG @snddlZddlZddlmZddlmZddlmZddl m Z dddZ d d Z d d Z d dZddZdS)N)make_np) make_grid)tf) EmbeddingInfocCs|sdd|D}n0t|t|dks0tddd|g|D}tjd|d}tjj|}|j ||d|dd dS) NcSsg|] }t|qSstr).0xrrF/tmp/pip-unpacked-wheel-gikjz4vx/torch/utils/tensorboard/_embedding.py szmake_tsv..rz@len of header must be equal to the number of columns in metadatacSs g|]}ddd|DqS) css|]}t|VqdS)Nr)r errr sz&make_tsv...)join)r lrrr r s metadata.tsvTZ binary_mode) lenAssertionErrorrcompatas_bytesriogfileget_filesystemwrite)metadata save_pathZmetadata_headerZmetadata_bytesfsrrr make_tsv sr!c Csddlm}ddlm}tt|dd}tt ||d}t |j d|j ddf}| ddd}||d|j dddddf<|t |d dd }|} |j| d d | } W5QRXtjj|} | j| |d | d ddS)Nr)Image)BytesIOg?)ZncolsrZPNG)format sprite.pngTr)ZPILr"rr#intmathceilsizerrnpzerosshapeZ transposeZ fromarrayZuint8Zclipsavegetvaluerrrrr) label_imgrr"r#ZnrowZarranged_img_CHWZarranged_augment_square_HWCZarranged_img_HWCZimbufZim_bytesr rrr make_sprites   r4cCs|t}d|t|d|_||d|_|dk rB||d|_|dk rx||d|j_ |jj | d| dg|S)Nz{}:{} tensors.tsvrr(r%r$) rr'r zfillZ tensor_namerZ tensor_pathZ metadata_pathZspriteZ image_pathZsingle_image_dimextendr,)rr2ZfilesyssubdirZ global_steptaginforrr get_embedding_info1sr<cCs6tjj|}||d}|j|tj|dddS)Nzprojector_config.pbtxtTr)rrrrrrrr)rcontentsr Z config_pathrrr write_pbtxt=s r>c Csjtjj|}tjj||dd:}|D].}dd|D}|tjd|dq,W5QRXdS)Nr6wbcSsg|]}t|qSr)r item)r irrr r Gszmake_mat..rr) rrrrZGFilerrrr)Zmatlistrr fr rrr make_matCs rC)N)r*Znumpyr-Z _convert_npr_utilsrZtensorboard.compatrZ2tensorboard.plugins.projector.projector_config_pb2rr!r4r<r>rCrrrr s