这是一个方法,用于在TPU上拟合批量样本后整理结果。
参数说明:
batched_examples:包含图像和标签等信息的特征(字典)和标签(字典)元组。predictions:形状为(bsz, h, w, 2)的int32类型预测结果。training:布尔值类型,表示是否处于训练模式。
函数流程说明:
- 获取特征中的条件映射cond_map,如果不存在,则创建与num_frames相同形状的全0张量作为cond_map。
- 将处理后的数据打包成一个元组返回给
postprocess_cpu()方法进行CPU模式下的后处理。具体而言,返回了如下7个元素:- examples[‘image’]:原始图像数据;
- cond_map:条件映射;
- examples[‘video_id’]:视频ID;
- examples[‘orig_image_size’]:原始图像大小;
- examples[‘unpadded_image_size’]:未填充过的图像大小;
- examples[‘num_frames’]:视频帧数;
- predictions:预测结果;
- examples[‘label_map’]:标签信息。




