ncnn_ssd

ssd.pytorch to ncnn C++ and android platform.

pytorch ssd fork from pytorch.ssd

软件环境

• windows 10

• visual studio 2017

• ncnn （cpu or vulkan）

• opencv 3.0

• VulkanSDK-1.1.92.1(option)

  **Download ThirdParty(opencv ncnn vulkan) , please check here BaiduDrive 提取码：n3jl , finish ,**move ThirdParty file to $$

demo

• load model

  char parmFile[] = "../models/ssd/ssd_vgg300.param";
  char binFile[] = "../models/ssd/ssd_vgg300.bin";
  SSDDetector ssdDet;
  ssdDet.loadModel(parmFile, binFile);

  ​

• 设置线程

  ssdDet.setNumTheads(1);

  ​

• forward

  const float fMean[3] = { 104.0f, 117.0f, 123.0f };
  double start_time = ncnn::get_current_time();
  ssdDet.detector(img.data, img.cols, img.rows, fMean, NULL, 0);
  double end_time = ncnn::get_current_time();

Model

• move L2Norm layer and finetuen mAP=74.9 ,download ncnn model ,please check here [BaiduDrive] ,提取码：nd9w

TODO

• Support gpu
• Ssd support android ,check here : https://github.com/jmu201521121021/ncnn-android-ssd

实验结果