BoxDecodeType.h File

Typed decode-family selection for BoxDecode stages. More...

Included Headers

#include <cstdint>

Namespaces Index

namespace	simaai
namespace	neat

Description

Typed decode-family selection for BoxDecode stages.

The BoxDecodeType enum picks the head-decoding contract used by the SimaBoxDecode postprocess stage: which YOLO variant, DETR, EffDet, RCNN stage-1 or CenterNet head layout the planner should expect. The companion BoxDecodeTypeOption enum selects the tensor-packing variant within a family. Helpers in this header stringify the enums and summarise each decode family's tensor contract for diagnostics and docs.

See Also

nodes/sima/SimaBoxDecode.h for the consuming postprocess node.

See Also

DetectionTypes.h for the decoded detection payload.

File Listing

The file content with the documentation metadata removed is:

1

16#pragma once

17

18#include <cstdint>

19

20namespace simaai::neat {

21

33enum class BoxDecodeType : std::int32_t {

34 // Internal sentinel; runtime decode requires a concrete value.

35 Unspecified = 0,

36 // YOLO-family (generic token).

37 Yolo = 1,

38 YoloV5 = 2,

39 YoloV5Seg = 3,

40 YoloV7 = 4,

41 YoloV7Seg = 5,

42 YoloV8 = 6,

43 YoloV8Seg = 7,

44 YoloV8Pose = 8,

45 YoloV9 = 9,

46 YoloV9Seg = 10,

47 YoloV10 = 11,

48 YoloV10Seg = 12,

49 Detr = 13,

50 EffDet = 14,

51 RcnnStage1 = 15,

52 Centernet = 16,

53 YoloV26 = 17,

54};

55

66enum class BoxDecodeTypeOption : std::int32_t {

67 Auto = 0,

68 PackedPerHead = 1,

69 InterleavedByHead = 2,

70 GroupedByRole = 3,

71 Split3Interleaved = 4,

72 Split3Grouped = 5,

73 InterleavedByHeadProbability = 6,

74 InterleavedByHeadLogit = 7,

75 GroupedByRoleProbability = 8,

76 GroupedByRoleLogit = 9,

77};

78

80constexpr const char* box_decode_type_token(BoxDecodeType type) {

81 switch (type) {

82 case BoxDecodeType::Yolo:

83 return "yolo";

84 case BoxDecodeType::YoloV5:

85 return "yolov5";

86 case BoxDecodeType::YoloV5Seg:

87 return "yolov5-seg";

88 case BoxDecodeType::YoloV7:

89 return "yolov7";

90 case BoxDecodeType::YoloV7Seg:

91 return "yolov7-seg";

92 case BoxDecodeType::YoloV8:

93 return "yolov8";

94 case BoxDecodeType::YoloV8Seg:

95 return "yolov8-seg";

96 case BoxDecodeType::YoloV8Pose:

97 return "yolov8-pose";

98 case BoxDecodeType::YoloV9:

99 return "yolov9";

100 case BoxDecodeType::YoloV9Seg:

101 return "yolov9-seg";

102 case BoxDecodeType::YoloV10:

103 return "yolov10";

104 case BoxDecodeType::YoloV10Seg:

105 return "yolov10-seg";

106 case BoxDecodeType::YoloV26:

107 return "yolo26";

108 case BoxDecodeType::Detr:

109 return "detr";

110 case BoxDecodeType::EffDet:

111 return "effdet";

112 case BoxDecodeType::RcnnStage1:

113 return "rcnn-stage1";

114 case BoxDecodeType::Centernet:

115 return "centernet";

116 case BoxDecodeType::Unspecified:

117 default:

118 return "unspecified";

119 }

120}

121

123constexpr const char* box_decode_type_option_token(BoxDecodeTypeOption option) {

124 switch (option) {

125 case BoxDecodeTypeOption::Auto:

126 return "auto";

127 case BoxDecodeTypeOption::PackedPerHead:

128 return "packed-per-head";

129 case BoxDecodeTypeOption::InterleavedByHead:

130 return "interleaved-by-head";

131 case BoxDecodeTypeOption::GroupedByRole:

132 return "grouped-by-role";

133 case BoxDecodeTypeOption::Split3Interleaved:

134 return "split3-interleaved";

135 case BoxDecodeTypeOption::Split3Grouped:

136 return "split3-grouped";

137 case BoxDecodeTypeOption::InterleavedByHeadProbability:

138 return "interleaved-by-head-probability";

139 case BoxDecodeTypeOption::InterleavedByHeadLogit:

140 return "interleaved-by-head-logit";

141 case BoxDecodeTypeOption::GroupedByRoleProbability:

142 return "grouped-by-role-probability";

143 case BoxDecodeTypeOption::GroupedByRoleLogit:

144 return "grouped-by-role-logit";

145 default:

146 return "auto";

147 }

148}

149

151constexpr bool box_decode_type_is_yolo_family(BoxDecodeType type) {

152 switch (type) {

153 case BoxDecodeType::Yolo:

154 case BoxDecodeType::YoloV5:

155 case BoxDecodeType::YoloV5Seg:

156 case BoxDecodeType::YoloV7:

157 case BoxDecodeType::YoloV7Seg:

158 case BoxDecodeType::YoloV8:

159 case BoxDecodeType::YoloV8Seg:

160 case BoxDecodeType::YoloV8Pose:

161 case BoxDecodeType::YoloV9:

162 case BoxDecodeType::YoloV9Seg:

163 case BoxDecodeType::YoloV10:

164 case BoxDecodeType::YoloV10Seg:

165 case BoxDecodeType::YoloV26:

166 return true;

167 case BoxDecodeType::Detr:

168 case BoxDecodeType::EffDet:

169 case BoxDecodeType::RcnnStage1:

170 case BoxDecodeType::Centernet:

171 case BoxDecodeType::Unspecified:

172 default:

173 return false;

174 }

175}

176

178constexpr bool box_decode_type_is_segmentation(BoxDecodeType type) {

179 switch (type) {

180 case BoxDecodeType::YoloV5Seg:

181 case BoxDecodeType::YoloV7Seg:

182 case BoxDecodeType::YoloV8Seg:

183 case BoxDecodeType::YoloV9Seg:

184 case BoxDecodeType::YoloV10Seg:

185 return true;

186 case BoxDecodeType::Yolo:

187 case BoxDecodeType::YoloV5:

188 case BoxDecodeType::YoloV7:

189 case BoxDecodeType::YoloV8:

190 case BoxDecodeType::YoloV8Pose:

191 case BoxDecodeType::YoloV9:

192 case BoxDecodeType::YoloV10:

193 case BoxDecodeType::YoloV26:

194 case BoxDecodeType::Detr:

195 case BoxDecodeType::EffDet:

196 case BoxDecodeType::RcnnStage1:

197 case BoxDecodeType::Centernet:

198 case BoxDecodeType::Unspecified:

199 default:

200 return false;

201 }

202}

203

205constexpr const char* box_decode_type_contract_summary(BoxDecodeType type) {

206 switch (type) {

207 case BoxDecodeType::Yolo:

208 case BoxDecodeType::YoloV5:

209 case BoxDecodeType::YoloV7:

210 case BoxDecodeType::YoloV8:

211 case BoxDecodeType::YoloV8Pose:

212 case BoxDecodeType::YoloV9:

213 case BoxDecodeType::YoloV10:

214 return "YOLO tensor contract: decoupled class heads (>4 channels, repeated across heads) or "

215 "packed heads (depth=3*(num_classes+5), consistent across heads).";

216 case BoxDecodeType::YoloV26:

217 return "YOLO26 tensor contract: grouped raw l/t/r/b bbox heads (4 channels) paired "

218 "with repeated class-score heads (>4 channels).";

219 case BoxDecodeType::YoloV5Seg:

220 case BoxDecodeType::YoloV7Seg:

221 case BoxDecodeType::YoloV8Seg:

222 case BoxDecodeType::YoloV9Seg:

223 case BoxDecodeType::YoloV10Seg:

224 return "YOLO-seg contract: same class-depth rules as YOLO plus segmentation decode path.";

225 case BoxDecodeType::Detr:

226 return "DETR contract: class channels inferred from max depth across heads; max depth must be "

227 ">4.";

228 case BoxDecodeType::EffDet:

229 case BoxDecodeType::RcnnStage1:

230 case BoxDecodeType::Centernet:

231 return "Non-YOLO fallback contract: class channels inferred from max depth across heads; "

232 "max depth must be >4.";

233 case BoxDecodeType::Unspecified:

234 default:

235 return "Unset decode type; runtime decode must fail fast until a concrete decode family is "

236 "selected.";

237 }

238}

239

240} // namespace simaai::neat

---

Generated via doxygen2docusaurus 2.0.0 by Doxygen 1.9.8.