| @@ -166,6 +166,38 @@ void shufflenet_run(const ncnn::Net& net) | |||||
| ex.extract("fc1000", out); | 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) | void googlenet_init(ncnn::Net& net) | ||||
| { | { | ||||
| net.load_param("googlenet.param"); | net.load_param("googlenet.param"); | ||||
| @@ -278,36 +310,20 @@ void mobilenet_yolo_run(const ncnn::Net& net) | |||||
| ex.extract("detection_out", out); | 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::Extractor ex = net.create_extractor(); | ||||
| ncnn::Mat in(224, 224, 3); | |||||
| ncnn::Mat in(416, 416, 3); | |||||
| ex.input("data", in); | ex.input("data", in); | ||||
| ncnn::Mat out; | ncnn::Mat out; | ||||
| ex.extract("prob", out); | |||||
| ex.extract("detection_out", out); | |||||
| } | } | ||||
| int main(int argc, char** argv) | 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-yolo", mobilenet_yolo_init, mobilenet_yolo_run); | ||||
| benchmark("mobilenet-yolov3", mobilenet_yolov3_init, mobilenet_yolov3_run); | |||||
| return 0; | return 0; | ||||
| } | } | ||||