nexar_large_speed.py 文件源码

def decode_jpeg(self, image_buffer, scope=None):
        if FLAGS.fast_jpeg_decode == "pyfunc":
            print("using ctypes jpeg decode...")
            lib_jpeg = ctypes.cdll.LoadLibrary('./data_providers/decode_jpeg_memory/decode_memory.so')
            global ctypes_jpeg
            ctypes_jpeg = lib_jpeg.decode_jpeg_memory_turbo
            return self.decode_jpeg_python(image_buffer, scope)
        elif FLAGS.fast_jpeg_decode=="tf":
            print("using tensorflow binary libjpeg turbo")
            decode_jpeg_batch = tf.load_op_library(
                './data_providers/decode_jpeg_memory/decode_jpeg_batch.so').decode_jpeg_batch
            assert( FLAGS.decode_downsample_factor == 1 )
            ans = decode_jpeg_batch(image_buffer, FLAGS.IM_HEIGHT, FLAGS.IM_WIDTH)
            ans.set_shape([FLAGS.FRAMES_IN_SEG // FLAGS.temporal_downsample_factor,
                           FLAGS.IM_HEIGHT, FLAGS.IM_WIDTH, 3])
            return ans
        else:
            return self.decode_jpeg_original(image_buffer, scope)