How To Question: Change Background to Black

[kaisark]

I tested with the test_on_single_video.py script and everything is working, but I wanted to know what is the quickest way to change the background to black so that only the keypoints are seen? Does vis.py (vis_one_image_opencv) need to be modified?

Example below is with openpose...

Thanks.

[kaisark]

I found a solution. I added a line of code to _vis_single_frame in the test_on_single_video.py script.

def _vis_single_frame(im, cls_boxes_i, cls_segms_i, cls_keyps_i, cls_tracks_i, thresh): im = np.zeros(im.shape) #kkhatak - line of code added to make background black ### res = vis_utils.vis_one_image_opencv( im, cls_boxes_i, segms=cls_segms_i, keypoints=cls_keyps_i, tracks=cls_tracks_i, thresh=thresh, show_box=False, show_class=False, linewidth = 6) if res is None: return im return res