(self, is_training=False, size=224, interpolation='bicubic')
| 204 | class TfPreprocessTransform: |
| 205 | |
| 206 | def __init__(self, is_training=False, size=224, interpolation='bicubic'): |
| 207 | self.is_training = is_training |
| 208 | self.size = size[0] if isinstance(size, tuple) else size |
| 209 | self.interpolation = interpolation |
| 210 | self._image_bytes = None |
| 211 | self.process_image = self._build_tf_graph() |
| 212 | self.sess = None |
| 213 | |
| 214 | def _build_tf_graph(self): |
| 215 | with tf.device('/cpu:0'): |
nothing calls this directly
no test coverage detected