中文 | English
SegmentationResult代码定义在fastdeploy/vision/common/result.h
中,用于表明图像中每个像素预测出来的分割类别和分割类别的概率值。
fastdeploy::vision::SegmentationResult
struct SegmentationResult {
std::vector<uint8_t> label_map;
std::vector<float> score_map;
std::vector<int64_t> shape;
bool contain_score_map = false;
void Clear();
std::string Str();
};
- label_map: 成员变量,表示单张图片每个像素点的分割类别,
label_map.size()
表示图片像素点的个数 - score_map: 成员变量,与label_map一一对应的所预测的分割类别概率值(当导出模型时指定
--output_op argmax
)或者经过softmax归一化化后的概率值(当导出模型时指定--output_op softmax
或者导出模型时指定--output_op none
同时模型初始化的时候设置模型类成员属性apply_softmax=True
) - shape: 成员变量,表示输出图片的shape,为H*W
- Clear(): 成员函数,用于清除结构体中存储的结果
- Str(): 成员函数,将结构体中的信息以字符串形式输出(用于Debug)
fastdeploy.vision.SegmentationResult
- label_map(list of int): 成员变量,表示单张图片每个像素点的分割类别
- score_map(list of float): 成员变量,与label_map一一对应的所预测的分割类别概率值(当导出模型时指定
--output_op argmax
)或者经过softmax归一化化后的概率值(当导出模型时指定--output_op softmax
或者导出模型时指定--output_op none
同时模型初始化的时候设置模型类成员属性apply_softmax=true
) - shape(list of int): 成员变量,表示输出图片的shape,为H*W