Custom自定义

1 模块导入&FLAG设置import torch.backends.cudnn as cudnnfrom torch.utils.data.distributed import DistributedSamplerimport argparseparser = argparse.ArgumentParser(description='Train segmentation network')parser.add_argument("--local_rank", type=int, d

参考文章：https://github.com/onnx/tutorials/blob/master/tutorials/OnnxTensorflowExport.ipynbhttps://www.jianshu.com/p/8ec3a6c9c453使用TensorFlow框架训练模型，然后导出为ONNX格式，一般需要经历以下几个步骤：训练（Training）转PB文件（Graph Freezing）模型格式转换（Model Conversion）训练（Training）.

代码链接：https://github.com/xingyizhou/CenterNet/blob/1085662179604dd4c2667e3159db5445a5f4ac76/src/lib/models/decode.py#L497代码位置：src/lib/models/decode.py代码注释def multi_pose_decode(heat, wh, kps, r...

原文地址：https://blog.youkuaiyun.com/jepco1/article/details/80864022 realsense的开发资料主要集中linux下，linux对图像的处理确实有很多优势，比如大名鼎鼎的TensorFlow，OrbSlam2都是很难移植到windows下的。一些时候，开发...

问题：自己写了个dataloader，为了部署方便，用OpenCV的接口进行数据读取，而没有用PIL，代码大致如下： def __getitem__(self, idx): sample = self.samples[idx] img = cv2.imread(sample[0]) img = cv2.resize(img, tup...

最近在做一个细粒度识别的项目，具体而言是为了做一个特定场景的车辆/车种检测，因为摄像头角度问题，大多时候只有车辆的一部分处于画面内，所以没有走检测的方式，而是尝试了一下通过各种数据增强（主要是裁剪）来指导网络对不同车种车辆的各个部位进行学习，从而指导车种分类以及有无车辆检测，最终达到监控车辆（车种）的目的。但是因为数据集不大，尽管模型收敛很好，但心里还是没底，担心是过拟合。于是想到可以可视化一...

『写在前面』以CTC Beam search decoder为例，简单整理一下TensorFlow实现自定义Op的操作流程。基本的流程1. 定义Op接口#include "tensorflow/core/framework/op.h"REGISTER_OP("Custom") .Input("custom_input: int32") .Outpu...

链接PCL官网：http://www.pointclouds.org/PCL GitHub：https://github.com/PointCloudLibrary/pclpython-pcl GitHub：https://github.com/strawlab/python-pclPCL编译安装1、下载pcl-pcl-1.9.1.zip，解压；2、mkdir build ...

在移动端和嵌入式平台上使用NCNN框架调用caffe模型。NCNN编译安装下载链接：https://github.com/Tencent/ncnn/archive/20181228.zip Build from sourcemkdir build && cd build# 在cmake之前检查一下CMakeLists.txt末尾是否打开了target-tools...

参考文章：https://www.zouyesheng.com/python-module-c.html    整体流程处理从Python中传入的参数 使用C/C++实现功能逻辑 将功能逻辑产生的返回值包装成Python所需的格式 注册函数 注册模块名 编译具体步骤一、引入头文件#include <Python.h>该文件一般位于python安装目...

1 依赖包版本说明       表格中列举了我试验成功的所有依赖包及其对应版本号，部分依赖包可能和官方文档中所推荐的不同。经本人验证，caffe编译过程中所需的依赖包版本相互之间有严重的依赖关系，当然其他版本可能也能正常运行，但如果想节省时间的话，不妨试试以下这些吧~ Package Version Description Prot...

『写在前面』 关于tf，一种常见的使用方式是：在线下使用TensorFlow的Python Binding搭建和训练模型，然后利用freeze_graph工具等工具输出*.pb文件（或使用tf.train.Saver保存成*.meta文件和checkpoint文件）,最后在线上使用C++调用预训练好的模型进行inference。之前趟过一遍TF1.3.1，坑很多，有点后悔当时没有记录，主要还是在...