(self, image)
| 210 | self.sess.close() |
| 211 | |
| 212 | def get_net_out(self, image): |
| 213 | begin = time.time() |
| 214 | if self.use_frozen_net: |
| 215 | out_var = 'prefix/model/add_2' |
| 216 | else: |
| 217 | out_var = self.net.p |
| 218 | net_out = self.sess.run(out_var, feed_dict={ |
| 219 | self.net_input_placeholder: image.reshape(1, *image.shape),}) |
| 220 | # print(net_out) |
| 221 | end = time.time() |
| 222 | log.debug('inference time %s', end - begin) |
| 223 | return net_out |
| 224 | |
| 225 | def preprocess_obz(self, obz): |
| 226 | for camera in obz['cameras']: |