在自己的程序中使用caffe的CPU版本库

#include <caffe/caffe.hpp>
#include <opencv2/core/core.hpp>
#include <opencv2/highgui/highgui.hpp>
#include <opencv2/imgproc/imgproc.hpp>
#include <string>
#include <iostream>
#include <sys/time.h>

using namespace caffe;
// using namespace std;

// std::string CLASSES[81] = { "__background__",
// "plastic_bag", "paper", "car", "motorcycle",
// "airplane", "bus", "train", "truck", "boat",
// "traffic light", "fire hydrant", "stop sign", "parking meter",
// "bench", "bird", "cat",
// "dog", "horse", "sheep", "cow",
// "elephant", "bear", "zebra", "giraffe" ,
// "backpack", "umbrella", "handbag", "tie" ,
// "suitcase", "frisbee", "skis", "snowboard" ,
// "sports ball", "kite", "baseball bat", "baseball glove" ,
// "skateboard", "surfboard", "tennis racket", "bottle" ,
// "wine glass", "cup", "fork", "knife" ,
// "spoon", "bowl", "banana", "apple" ,
// "sandwich", "orange", "broccoli", "carrot" ,
// "hot dog", "pizza", "donut", "cake" ,
// "chair", "couch", "potted plant", "bed" ,
// "dining table", "toilet", "tv", "laptop" ,
// "mouse", "remote", "keyboard", "cell phone" ,
// "microwave", "oven", "toaster", "sink" ,
// "refrigerator", "book", "clock", "vase" ,
// "scissors", "teddy bear", "hair drier", "toothbrush" ,
// };

std::string CLASSES[81] = {
   
   
    "__background__",
    "aeroplane",
    "bicycle",
    "bird",
    "boat",
    "bottle",
    "bus",
    "car",
    "cat",
    "chair",
    "cow",
    "diningtable",
    "dog",
    "horse",
    "motorbike",
    "person",
    "pottedplant",
    "sheep",
    "sofa",
    "train",
    "tvmonitor"};

int main()
{
   
   
    // std::string model_file("/home/lwd/code/dl/ssd/caffe/examples/MobileNet-SSD/voc/compare.prototxt");
    // std::string weights_file("/home/lwd/code/dl/ssd/caffe/snapshot/mobilenet_iter_40000.caffemodel");
    std::string weights_file("/home/lwd/ssddetect/MobileNetSSD_deploy.caffemodel");
    std::string model_file("/home/lwd/ssddetect/MobileNetSSD_deploy.prototxt");
    cv::Mat mean_;

    // set net
    Caffe::set_mode(Caffe::CPU);
    shared_ptr<Net<float>> net_
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

刀么克瑟拉莫

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值