def _setup_input(self): self.inputs = tf.placeholder( tf.float32, (None, self.config.size[0], self.config.size[1], 3)) self.data_batches = gray_scale( self.inputs) if self.config.gray_scale else self.inputs
def _setup_input(self): self.inputs = tf.placeholder( tf.float32, (None, self.config.size[0], self.config.size[1], 3), name=YOLOToExportModel.INPUT_NODE_NAME) self.data_batches = gray_scale( self.inputs) if self.config.gray_scale else self.inputs