cuda
小涵涵
这个作者很懒,什么都没留下…
展开
-
ckpt-pb-onnx-trt
ckpt pb onnx trtckptpbonnxtrtbug思考 ckpt 这应该是ckpt转换成自己网络名称的操作 import argparse import tensorflow as tf from core.yolov3 import YOLOV3 from core.config import cfg parser = argparse.ArgumentParser() parser.add_argument("--train_from_coco", action='store_true'原创 2020-08-02 14:25:41 · 1301 阅读 · 1 评论 -
caffe blob
caffe blob类函数CanonicalAxisIndex(int axis_index)LegacyShape(int index)TIP 类函数 返回形状信息bcwh inline const vector<int>& shape() const { return shape_; } 返回index维度的大小 inline int shape(int index) const { return shape_[CanonicalAxisIndex(index)];原创 2020-07-27 15:25:04 · 184 阅读 · 0 评论 -
opencv gpu
opencv gpu普通的copy操作opencv cudacuda 实现 resize croptip 普通的copy操作 cv::Mat src = imread(szfile); cv::Mat dst; cv::gpu::GpuMat mat1(src); mat1.upload(src); mat1.download(dst); cpu和gpu之间的copy操作,upload()用于将host数据传入到device,download用于将数据从device传回host. opencv cuda原创 2020-07-23 17:16:35 · 468 阅读 · 0 评论 -
tensorrt cuda函数编写
cu文件的编写.h文件的编写 .h文件的编写 #ifndef TINYTRT_DECODE_H #define TINYTRT_DECODE_H #include "Trt.h" #include "utils.h" #include "YoloLayerPlugin/YoloLayerPlugin.hpp" using namespace std; struct Bbox { int left, right, top, bottom; int clsId; float score;原创 2020-07-16 14:08:27 · 441 阅读 · 0 评论