Browse Source

add the converted models

tags/20200106
nihuini 6 years ago
parent
commit
ca6d4c8726
10 changed files with 11 additions and 1 deletions
  1. +2
    -1
      examples/fasterrcnn.cpp
  2. +1
    -0
      examples/mobilenetssd.cpp
  3. +1
    -0
      examples/mobilenetv2ssdlite.cpp
  4. +1
    -0
      examples/peleenetssd_seg.cpp
  5. +1
    -0
      examples/retinaface.cpp
  6. +1
    -0
      examples/simplepose.cpp
  7. +1
    -0
      examples/squeezenet.cpp
  8. +1
    -0
      examples/squeezenetssd.cpp
  9. +1
    -0
      examples/yolov2.cpp
  10. +1
    -0
      examples/yolov3.cpp

+ 2
- 1
examples/fasterrcnn.cpp View File

@@ -114,7 +114,8 @@ static int detect_fasterrcnn(const cv::Mat& bgr, std::vector<Object>& objects)
// py-faster-rcnn/models/pascal_voc/ZF/faster_rcnn_alt_opt/faster_rcnn_test.pt
// https://dl.dropboxusercontent.com/s/o6ii098bu51d139/faster_rcnn_models.tgz?dl=0
// ZF_faster_rcnn_final.caffemodel
fasterrcnn.load_param("ZF_faster_rcnn_final.proto");
// the ncnn model https://github.com/nihui/ncnn-assets/tree/master/models
fasterrcnn.load_param("ZF_faster_rcnn_final.param");
fasterrcnn.load_model("ZF_faster_rcnn_final.bin");

// hyper parameters taken from


+ 1
- 0
examples/mobilenetssd.cpp View File

@@ -41,6 +41,7 @@ static int detect_mobilenet(const cv::Mat& bgr, std::vector<Object>& objects)

// model is converted from https://github.com/chuanqi305/MobileNet-SSD
// and can be downloaded from https://drive.google.com/open?id=0ByaKLD9QaPtucWk0Y0dha1VVY0U
// the ncnn model https://github.com/nihui/ncnn-assets/tree/master/models
mobilenet.load_param("mobilenet_ssd_voc_ncnn.param");
mobilenet.load_model("mobilenet_ssd_voc_ncnn.bin");



+ 1
- 0
examples/mobilenetv2ssdlite.cpp View File

@@ -46,6 +46,7 @@ static int detect_mobilenetv2(const cv::Mat& bgr, std::vector<Object>& objects)

// original pretrained model from https://github.com/chuanqi305/MobileNetv2-SSDLite
// https://github.com/chuanqi305/MobileNetv2-SSDLite/blob/master/ssdlite/voc/deploy.prototxt
// the ncnn model https://github.com/nihui/ncnn-assets/tree/master/models
mobilenetv2.load_param("mobilenetv2_ssdlite_voc.param");
mobilenetv2.load_model("mobilenetv2_ssdlite_voc.bin");



+ 1
- 0
examples/peleenetssd_seg.cpp View File

@@ -41,6 +41,7 @@ static int detect_peleenet(const cv::Mat& bgr, std::vector<Object>& objects,ncnn

// model is converted from https://github.com/eric612/MobileNet-YOLO
// and can be downloaded from https://drive.google.com/open?id=1Wt6jKv13sBRMHgrGAJYlOlRF-o80pC0g
// the ncnn model https://github.com/nihui/ncnn-assets/tree/master/models
peleenet.load_param("pelee.param");
peleenet.load_model("pelee.bin");



+ 1
- 0
examples/retinaface.cpp View File

@@ -248,6 +248,7 @@ static int detect_retinaface(const cv::Mat& bgr, std::vector<FaceObject>& faceob
// model is converted from
// https://github.com/deepinsight/insightface/tree/master/RetinaFace#retinaface-pretrained-models
// https://github.com/deepinsight/insightface/issues/669
// the ncnn model https://github.com/nihui/ncnn-assets/tree/master/models
// retinaface.load_param("retinaface-R50.param");
// retinaface.load_model("retinaface-R50.bin");
retinaface.load_param("mnet.25-opt.param");


+ 1
- 0
examples/simplepose.cpp View File

@@ -47,6 +47,7 @@ static int detect_posenet(const cv::Mat& bgr, std::vector<KeyPoint>& keypoints)
// pose_net.hybridize()
// pose_net.export('pose')
// then mxnet2ncnn
// the ncnn model https://github.com/nihui/ncnn-assets/tree/master/models
posenet.load_param("pose.param");
posenet.load_model("pose.bin");



+ 1
- 0
examples/squeezenet.cpp View File

@@ -32,6 +32,7 @@ static int detect_squeezenet(const cv::Mat& bgr, std::vector<float>& cls_scores)
squeezenet.opt.use_vulkan_compute = true;
#endif // NCNN_VULKAN

// the ncnn model https://github.com/nihui/ncnn-assets/tree/master/models
squeezenet.load_param("squeezenet_v1.1.param");
squeezenet.load_model("squeezenet_v1.1.bin");



+ 1
- 0
examples/squeezenetssd.cpp View File

@@ -42,6 +42,7 @@ static int detect_squeezenet(const cv::Mat& bgr, std::vector<Object>& objects)
// original pretrained model from https://github.com/chuanqi305/SqueezeNet-SSD
// squeezenet_ssd_voc_deploy.prototxt
// https://drive.google.com/open?id=0B3gersZ2cHIxdGpyZlZnbEQ5Snc
// the ncnn model https://github.com/nihui/ncnn-assets/tree/master/models
squeezenet.load_param("squeezenet_ssd_voc.param");
squeezenet.load_model("squeezenet_ssd_voc.bin");



+ 1
- 0
examples/yolov2.cpp View File

@@ -42,6 +42,7 @@ static int detect_yolov2(const cv::Mat& bgr, std::vector<Object>& objects)
// original pretrained model from https://github.com/eric612/MobileNet-YOLO
// https://github.com/eric612/MobileNet-YOLO/blob/master/models/yolov2/mobilenet_yolo_deploy.prototxt
// https://github.com/eric612/MobileNet-YOLO/blob/master/models/yolov2/mobilenet_yolo_deploy_iter_80000.caffemodel
// the ncnn model https://github.com/nihui/ncnn-assets/tree/master/models
yolov2.load_param("mobilenet_yolo.param");
yolov2.load_model("mobilenet_yolo.bin");



+ 1
- 0
examples/yolov3.cpp View File

@@ -42,6 +42,7 @@ static int detect_yolov3(const cv::Mat& bgr, std::vector<Object>& objects)
// original pretrained model from https://github.com/eric612/MobileNet-YOLO
// param : https://drive.google.com/open?id=1V9oKHP6G6XvXZqhZbzNKL6FI_clRWdC-
// bin : https://drive.google.com/open?id=1DBcuFCr-856z3FRQznWL_S5h-Aj3RawA
// the ncnn model https://github.com/nihui/ncnn-assets/tree/master/models
yolov3.load_param("mobilenetv2_yolov3.param");
yolov3.load_model("mobilenetv2_yolov3.bin");



Loading…
Cancel
Save