| 241 | } |
| 242 | |
| 243 | void 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 |