MCPcopy Create free account
hub / github.com/FeiYull/TensorRT-Alpha / reset

Method reset

utils/yolo.cpp:243–250  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

241}
242
243void yolo::YOLO::reset()
244{
245 CHECK(cudaMemset(m_output_objects_device, 0, sizeof(float) * m_param.batch_size * (1 + 7 * m_param.topK)));
246 for (size_t bi = 0; bi < m_param.batch_size; bi++)
247 {
248 m_objectss[bi].clear();
249 }
250}
251
252
253

Callers 1

taskFunction · 0.45

Calls

no outgoing calls

Tested by

no test coverage detected