Exemple #1
0
 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
Exemple #2
0
 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