From 884e4f76ac80da5a1c2a4d35ffca2a7e5b0e553a Mon Sep 17 00:00:00 2001
From: nihuini <nihuini@tencent.com>
Date: Fri, 14 Dec 2018 19:43:05 +0800
Subject: [PATCH] add mobilenet-yolov3 benchmark

---
 benchmark/benchncnn.cpp | 60 ++++++++++++++++++++++++++---------------
 1 file changed, 39 insertions(+), 21 deletions(-)

diff --git a/benchmark/benchncnn.cpp b/benchmark/benchncnn.cpp
index fee83e845..0ce9c6c8e 100644
--- a/benchmark/benchncnn.cpp
+++ b/benchmark/benchncnn.cpp
@@ -166,6 +166,38 @@ void shufflenet_run(const ncnn::Net& net)
     ex.extract("fc1000", out);
 }
 
+void mnasnet_init(ncnn::Net& net)
+{
+    net.load_param("mnasnet.param");
+}
+
+void mnasnet_run(const ncnn::Net& net)
+{
+    ncnn::Extractor ex = net.create_extractor();
+
+    ncnn::Mat in(224, 224, 3);
+    ex.input("data", in);
+
+    ncnn::Mat out;
+    ex.extract("prob", out);
+}
+
+void proxylessnasnet_init(ncnn::Net& net)
+{
+    net.load_param("proxylessnasnet.param");
+}
+
+void proxylessnasnet_run(const ncnn::Net& net)
+{
+    ncnn::Extractor ex = net.create_extractor();
+
+    ncnn::Mat in(224, 224, 3);
+    ex.input("data", in);
+
+    ncnn::Mat out;
+    ex.extract("prob", out);
+}
+
 void googlenet_init(ncnn::Net& net)
 {
     net.load_param("googlenet.param");
@@ -278,36 +310,20 @@ void mobilenet_yolo_run(const ncnn::Net& net)
     ex.extract("detection_out", out);
 }
 
-void mnasnet_init(ncnn::Net& net)
-{
-    net.load_param("mnasnet.param");
-}
-
-void mnasnet_run(const ncnn::Net& net)
-{
-    ncnn::Extractor ex = net.create_extractor();
-
-    ncnn::Mat in(224, 224, 3);
-    ex.input("data", in);
-
-    ncnn::Mat out;
-    ex.extract("prob", out);
-}
-
-void proxylessnasnet_init(ncnn::Net& net)
+void mobilenet_yolov3_init(ncnn::Net& net)
 {
-    net.load_param("proxylessnasnet.param");
+    net.load_param("mobilenet_yolov3.param");
 }
 
-void proxylessnasnet_run(const ncnn::Net& net)
+void mobilenet_yolov3_run(const ncnn::Net& net)
 {
     ncnn::Extractor ex = net.create_extractor();
 
-    ncnn::Mat in(224, 224, 3);
+    ncnn::Mat in(416, 416, 3);
     ex.input("data", in);
 
     ncnn::Mat out;
-    ex.extract("prob", out);
+    ex.extract("detection_out", out);
 }
 
 int main(int argc, char** argv)
@@ -378,5 +394,7 @@ int main(int argc, char** argv)
 
     benchmark("mobilenet-yolo", mobilenet_yolo_init, mobilenet_yolo_run);
 
+    benchmark("mobilenet-yolov3", mobilenet_yolov3_init, mobilenet_yolov3_run);
+
     return 0;
 }