ó V¾÷Xc@`sÃdZddlmZddlmZddlmZddlmZddlmZddl m Z ddl m Z dd l m Z d „Z dd „Zdd „Zdd „Zdd„ZdS(sèMiscellaneous functions useful for nD-LSTM models. Some of these functions duplicate functionality in tfslim with slightly different interfaces. Tensors in this library generally have the shape (num_images, height, width, depth). i(tabsolute_import(tdivision(tprint_function(tlayers(tops(t array_ops(tmath_ops(t sparse_opscC`s|jƒjƒS(s)Get the shape of a tensor as an int list.(t get_shapetas_list(ttensor((sg/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/ndlstm/python/misc.pyt_shape"sc C`sYtj|d|gƒ;t|ƒ\}}}}tj|||||gƒSWdQXdS(s1Reduce images to vectors by combining all pixels.tPixelsAsVectorN(Rt name_scopeR Rtreshape(timagestscopet batch_sizetheighttwidthtdepth((sg/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/ndlstm/python/misc.pytpixels_as_vector'scC`s9tj|d|gƒtj|ddgƒSWdQXdS(s1Reduce images to vectors by averaging all pixels.t PoolAsVectoriiN(RR Rt reduce_mean(RR((sg/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/ndlstm/python/misc.pytpool_as_vector.sc C`s`tj|d|gƒBt|ƒ\}tj||ƒ}tj||dd|gƒSWdQXdS(sƒCompute 1-hot encodings for planes. Given a label, this computes a label image that contains 1 at all pixels in the plane corresponding to the target class and 0 in all other planes. Args: labels: (batch_size,) tensor num_classes: number of classes scope: optional scope name Returns: Tensor of shape (batch_size, 1, 1, num_classes) with a 1-hot encoding. t OneHotPlanesiN(RR R Rtone_hot_encodingRR(tlabelst num_classesRRtbatched((sg/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/ndlstm/python/misc.pytone_hot_planes4sc C`sùtj|d|gƒÛt|ƒ\}}}|dks@t‚tjtj|ddgƒƒ}t|ƒ\}}tjtjd|dƒddgƒ} tj | |gdƒ} t j | ||gddƒ} tj| |||gƒ} | SWdQXdS(seCompute 1-hot encodings for masks. Given a label image, this computes the one hot encoding at each pixel. Args: labels: (batch_size, width, height, 1) tensor containing labels. num_classes: number of classes scope: optional scope name Returns: Tensor of shape (batch_size, width, height, num_classes) with a 1-hot encoding. t OneHotMaskiiÿÿÿÿigð?gN( RR R tAssertionErrorRtto_int32RRtrangetconcatRtsparse_to_dense( RRRRRRt sparse_labelst sparse_sizet_tindicestconcatedt dense_resulttresult((sg/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/ndlstm/python/misc.pyt one_hot_maskIs!'   N(t__doc__t __future__RRRt'tensorflow.contrib.layers.python.layersRttensorflow.python.frameworkRttensorflow.python.opsRRRR tNoneRRRR,(((sg/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/ndlstm/python/misc.pyts