From 6674fc6f30cea746044cb713a537e159f0006385 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Thu, 7 Jun 2018 22:03:42 +0200 Subject: so limiting --- rpc/rpc.py | 20 ++++++++++++-------- 1 file changed, 12 insertions(+), 8 deletions(-) (limited to 'rpc/rpc.py') diff --git a/rpc/rpc.py b/rpc/rpc.py index 3ea7cd6..82db460 100644 --- a/rpc/rpc.py +++ b/rpc/rpc.py @@ -66,15 +66,19 @@ class CortexRPC(object): print('Get params call received!') return self._get_fn() - def send_pil_image(self, fn, meta, im): + def send_pil_image(self, fn, meta, im, format='JPEG'): output = BytesIO() - im = im.convert('RGB') - # im.save(output, format='JPEG') - im.save(output, format='PNG') - frame = output.getvalue() - output.close() - # self.send_frame(fn.replace('png', 'jpg'), meta, frame) - self.send_frame(fn, meta, frame) + if format == 'PNG': + im = im.convert('RGB') + im.save(output, format='PNG') + frame = output.getvalue() + output.close() + self.send_frame(fn, meta, frame) + else: + im.save(output, format='JPEG') + frame = output.getvalue() + output.close() + self.send_frame(fn.replace('png', 'jpg'), meta, frame) def send_status(self, key, value): if self.relay is not None: -- cgit v1.2.3-70-g09d2