diff --git a/embody_plot.py b/embody_plot.py index 169587bc34272d2f219323006a76058e81779dc0..1c8666da31b11c4336dbf049b8f27d98eae2f8bb 100644 --- a/embody_plot.py +++ b/embody_plot.py @@ -210,24 +210,22 @@ def plot_coordinates(coordinates, image_path=DEFAULT_IMAGE_PATH): except IndexError as err: app.logger.info(err) - point = ndimage.gaussian_filter(frame, sigma=5) - ax2.imshow(point, cmap='hot', interpolation='none') - # Try to send progress information to socket.io - if idx == 0: continue if round((idx / points_count) * 100) % (step * 5) == 0: try: - emit('progress', {'done':step * 5, 'from':100}) + emit('progress', {'done': step * 5, 'from': 100}) socketio.sleep(0.05) - except RuntimeError as err: - print(err) + except RuntimeError: continue step += 1 + point = ndimage.gaussian_filter(frame, sigma=5) + ax2.imshow(point, cmap='hot', interpolation='none') + image_mask = mpimg.imread(IMAGE_PATH_MASK) ax2.imshow(image_mask)