MCPcopy Create free account
hub / github.com/MegEngine/MegEngine / make_det

Method make_det

src/gopt/test/network.cpp:293–301  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

291} // namespace
292
293SymbolVarArray mgb::make_det(Network& network, size_t batch, DType out_dtype) {
294 SymbolVarArray outputs;
295 auto pyramids = make_pyramids(network, batch, out_dtype);
296 auto fpn_hv = fusion_pyramids_feature(network, pyramids, 16);
297 auto fpn_plate = fusion_pyramids_feature(network, pyramids, 16);
298 outputs.insert(outputs.end(), fpn_hv.begin(), fpn_hv.end());
299 outputs.insert(outputs.end(), fpn_plate.begin(), fpn_plate.end());
300 return outputs;
301}
302
303SymbolVar mgb::bottleneck(
304 Network& network, SymbolVar f, size_t input_channels, size_t channels, size_t t,

Callers

nothing calls this directly

Calls 5

make_pyramidsFunction · 0.85
fusion_pyramids_featureFunction · 0.85
insertMethod · 0.45
endMethod · 0.45
beginMethod · 0.45

Tested by

no test coverage detected