def flow2rgb(flow_map, max_value): flow_map_np = flow_map.detach().cpu().numpy() _, h, w = flow_map_np.shape flow_map_np[:,(flow_map_np[0] == 0) & (flow_map_np[1] == 0)] = float('nan') rgb_map = np.ones((3,h,w)).astype(np.float32) if max_value is not None: normalized_flow_map = flow_map_np / max_value else: normalized_flow_map = flow_map_np / (np.abs(flow_map_np).max()) rgb_map[0] += normalized_flow_map[0] rgb_map[1] -= 0.5*(normalized_flow_map[0] + normalized_flow_map[1]) rgb_map[2] += normalized_flow_map[1] return rgb_map.clip(0,1)
光流转RGB图的函数 flow2rgb
最新推荐文章于 2023-01-02 18:57:42 发布