Method EfficientDet

efficientdet/efficientdet.cpp:4–31

#include"NvInferPlugin.h"

EfficientDet::EfficientDet(const utils::InitParameter& param):m_param(param)
{
    // input
    m_input_src_device = nullptr;
    m_input_resize_device = nullptr;
    m_input_rgb_device = nullptr;
    CHECK(cudaMalloc(&m_input_src_device,    param.batch_size * 3 * param.src_h * param.src_w * sizeof(float)));
    CHECK(cudaMalloc(&m_input_resize_device, param.batch_size * 3 * param.dst_h * param.dst_w * sizeof(float)));
    CHECK(cudaMalloc(&m_input_rgb_device,    param.batch_size * 3 * param.dst_h * param.dst_w * sizeof(float)));
    // output
    m_output_num_device     = nullptr;  
    m_output_boxes_device   = nullptr;  
    m_output_scores_device  = nullptr;  
    m_output_classes_device = nullptr;
    m_output_num_host     = nullptr;    
    m_output_boxes_host   = nullptr;    
    m_output_scores_host  = nullptr;    
    m_output_classes_host = nullptr;    
    CHECK(cudaMalloc(&m_output_num_device,     param.batch_size * sizeof(int)));
    CHECK(cudaMalloc(&m_output_boxes_device,   param.batch_size * 1 * param.topK * 4 * sizeof(int)));
    CHECK(cudaMalloc(&m_output_scores_device,  param.batch_size * 1 * param.topK * sizeof(int)));
    CHECK(cudaMalloc(&m_output_classes_device, param.batch_size * 1 * param.topK * sizeof(int)));
    m_output_num_host     = new int[param.batch_size];
    m_output_boxes_host   = new int[param.batch_size * 1 * param.topK * 4];
    m_output_scores_host  = new int[param.batch_size * 1 * param.topK];
    m_output_classes_host = new int[param.batch_size * 1 * param.topK];
    m_objectss.resize(param.batch_size);
}

EfficientDet::~EfficientDet()
{