#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_