目标检测-RefineDet算法C++测试代码

个人微信公众号：AI研习图书馆，欢迎关注~

深度学习知识及资源分享，学习交流，共同进步~
相关链接

1. 论文地址
2. 项目地址
3. Refinedet文章翻译
4. Refinedet训练代码解析
5. Refinedet网络结构解析
6. 目标检测-RefineDet实现详细步骤
7. Refinedet论文总结
8. 目标检测-RefineDet训练脚本解析
9. 目标检测-RefineDet算法检测部分网络解析

1. 概述

Python版本的测试代码在相应的文件夹下已经写好了，这里根据对应的Python文件写了一个测试Demo。

2. 测试Demo

#include <stdio.h>
#include <stdlib.h>
#include <iostream>
#include <vector>
#include <map>
#include <memory>
#include <string>
#include <opencv2/opencv.hpp>
#include <caffe/blob.hpp>
#include <caffe/common.hpp>
#include <caffe/layer.hpp>
#include <caffe/net.hpp>

int main(int argc, char* argv[])
{
   
	//GPU设置
	int device_id = 0;
	caffe::Caffe::set_mode(caffe::Caffe::GPU);
	caffe::Caffe::SetDevice(device_id);
	std::cout << "Use GPU: " << device_id << std::endl;
	
	//网络均值
	std::vector<float> pixel_mean={
   104.0, 117.0, 123.0}