liangyuch commited on
Commit
809c0a6
β€’
1 Parent(s): e2d5627

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +12 -1
utils.py CHANGED
@@ -5,6 +5,7 @@ import numpy as np
5
  from detectron2.utils.colormap import colormap
6
  from detectron2.utils.visualizer import VisImage, Visualizer
7
  from mmdet.datasets.coco_panoptic import INSTANCE_OFFSET
 
8
 
9
 
10
  CLASSES = [
@@ -286,4 +287,14 @@ def show_result(img,
286
  if show:
287
  all_rel_vis.append(output_viz_graph)
288
 
289
- return all_rel_vis
 
 
 
 
 
 
 
 
 
 
 
5
  from detectron2.utils.colormap import colormap
6
  from detectron2.utils.visualizer import VisImage, Visualizer
7
  from mmdet.datasets.coco_panoptic import INSTANCE_OFFSET
8
+ from PIL import Image
9
 
10
 
11
  CLASSES = [
 
287
  if show:
288
  all_rel_vis.append(output_viz_graph)
289
 
290
+ return all_rel_vis
291
+
292
+
293
+ def make_gif(np_images):
294
+ frames = [Image.fromarray(numpy_image.astype('uint8'), 'RGB') for numpy_image in np_images]
295
+ # frames = [Image.open(image) for image in images]
296
+ frame_one = frames[0]
297
+ file_name = "top_rel.gif"
298
+ frame_one.save(file_name, format="GIF", append_images=frames,
299
+ save_all=True, duration=1000, loop=0)
300
+ return file_name