ab0902cd-CSDN博客

原创 nvidia docker安装

2. 安装nvidia docker。1.安装docker。3. gpu打进容器。

2024-01-09 11:32:32 500

原创 ubutnu qt.qpa.xcb: could not connect to display 1

解决办法：export DISPLAY：=1。

2023-07-13 16:23:14 199

原创多标签分类与binary_cross_entropy_with_logits

1. binary_cross_entropy_with_logits可用于多标签分类torch.nn.functional.binary_cross_entropy_with_logits等价于torch.nn.BCEWithLogitsLosstorch.nn.BCELoss+torch.nn.Sigmoid 等价于torch.nn.BCEWithLogitsLoss在pytorch中torch.nn.functional.binary_cross_entropy_with_log...

2022-04-25 21:14:48 10619

原创 tensorflow获取动态shape

tf.shape(a)和a.get_shape()比较相同点：都可以得到tensor a的尺寸不同点：tf.shape()中a 数据的类型可以是tensor, list, array　　　　a.get_shape()中a的数据类型只能是tensor,且返回的是一个元组（tuple）如果需要根据上一层的动态shape计算当前层的shape，该如何做呢x=tf.placeholder(tf.float32, shape=[None, 227,227,3] )但在运行的时候想知道None到

2022-03-31 15:16:29 2497

原创 GCC安装

1.下载GCC安装包，下载地址如下，选择需要的安装版本：https://mirrors.tuna.tsinghua.edu.cn/gnu/gcc/2.解压配置进入解压目录执行：./configure可能会遇到下面的问题：configure: error: Building GCC requires GMP 4.2+, MPFR 2.4.0+ and MPC 0.8.0+.Try the --with-gmp, --with-mpfr and/or --with-mpc optio.

2022-03-30 17:27:22 4428

原创 scp通过跳板机传输文件

ssh 通过代理机跳转内网proxyjump代理登录跳板机使用proxyjump时，可能会遇到连接中转服务器后报错：channel 0: open failed: administratively prohibited: open failedstdio forwarding failedkex_exchange_identification: Connection closed by remote host这是因为没有指定端口号，需要指定端口号还有可能遇到下面的问题：ssh: Co

2022-02-11 10:44:22 3762

原创 onnx重写输入和输出的维度

onnx模型输入是静态的，比如是1x3x960x960，但是想把输入改成动态输入，相应的输出也得改成动态，以下代码可以修改onnx模型的维度：import onnximport onnx.checkerimport onnx.utilsfrom onnx.tools import update_model_dims model = onnx.load('infer_rec.onnx')# 此处可以理解为获得了一个维度 “引用”，通过该 “引用“可以修改其对应的维度

2021-09-09 14:28:24 7710 4

原创 android cmake踩坑记录

ａｎｄｒｏｉｄ　ｓｔｕｄｉｏ最终正常编译ｓｏ库的环境配置如下：ａｎｄｒｏｉ　ｓｔｕｄｉｏ　版本：３.５ｇｒａｄｌｅ　ｖｅｒｓｉｏｎ　：5.6.3ｇｒａｄｌｅ　plugin　：３.４.１cmake配置：cmake { path "src/main/cpp/CMakeLists.txt" version "3.10.2"}ｎｄｋ：android-ndk-r16b问题1：Execution failed for task ':app:generateJsonMo

2021-04-14 15:42:53 480

原创感受野的计算

感受野的计算感受野（Receptive Field），指的是神经网络中神经元“看到的”输入区域，在卷积神经网络中，feature map上某个元素的计算受输入图像上某个区域的影响，这个区域即该元素的感受野。卷积神经网络中，越深层的神经元看到的输入区域越大，如下图所示，kernel size 均为3×33×3，stride均为1，绿色标记的是Layer2Layer2每个神经元看到的区域，黄色标记的是Layer3Layer3看到的区域，具体地，Layer2Layer2每个神经元可看到Layer1La..

2020-11-21 15:03:58 445

原创 C++多线程编程#pragma omp parallel

通常创建线程通过pthread_create来进行线程创建创建线程下面的程序，我们可以用它来创建一个 POSIX 线程：#include <pthread.h>pthread_create (thread, attr, start_routine, arg) 在这里，pthread_create创建一个新的线程，并让它可执行。下面是关于参数的说明：参数描述 thread 指向线程标识符指针。 attr 一个不透明的属性对象，可以被用来设置线程.

2020-09-24 11:52:26 22712

原创 JNI如何避免内存泄露

内存泄露主要是申请后没有及时释放导致的，谁申请谁释放是c++必须遵循的重要原则，malloc申请一般通过free进行释放，new申请的内存一般通过delete进行释放，如果不进行释放，内存会逐渐升高，最终无法申请到内存而导致内存泄露1. 哪些需要手动释放？不要手动释放（基本类型）： jint , jlong , jchar 需要手动释放（引用类型，数组家族）： jstring，jobject ，jobjectArray，jintArray ，jclass ，jmethodID

2020-09-24 11:19:07 381

原创 Rotate an image without cropping in OpenCV in C++

I'd like to rotate an image, but I can't obtain the rotated image without croppingMy original image:通常使用下面的代码进行crop:#include <opencv2/core/core.hpp>#include <opencv2/highgui/highgui.hpp>#include <opencv2/imgproc/imgproc.hpp&...

2020-09-14 15:32:49 363

原创如何使用addr2line命令定位程序出错代码所在行号

linux中或者android端使用addr2line命令定位代码出错的代码所在行，但有时候addr2line定位输出为??：？。并不能分析出哪行代码出了问题，例如，addr2line -e a.out 0x4005BDC命令。也可以使用valgrind工具运行此a.out可执行文件时获取此地址以查找内存泄漏。addr2line - 将地址转换为文件名和行号。addr2line常常会遇到下面的情况：$ addr2line -e a.out 0x400442 #offset in the `__

2020-08-06 16:32:41 6531

原创 pytorch模型转onnx

pytorch模型转其他的格式一般需要先转成通用的格式，onnx就是一种通用的格式，下面介绍转onnx的方法1.torch模型转onnx 直接通过torch官网：https://pytorch.org/docs/stable/onnx.html，提供的方法转onnximport torchvisiondummy_input = torch.randn(10, 3, 224, 224, device='cuda')model = torchvision.models.alexnet(pr.

2020-07-27 16:58:20 1226

原创 xgboost算法原理

1. XGBoost目标函数XGBoost的目标函数由训练损失和正则化项两部分组成，目标函数定义如下：第一项为损失函数，第二项为树的复杂度变量解释：（1）l 代表损失函数，常见的损失函数有：平方损失函数：逻辑回归损失函数：（2）yi'是第 i 个样本 xi 的预测值。由于XGBoost是一个加法模型，因此，预测得分是每棵树打分的累加之和。（3）将全部k棵树的复杂度进行求和，添加到目标函数中作为正则化项，用于防止模型过度拟合。2. 学习第t棵树在【1】中提到

2020-07-15 18:00:47 663

原创 docker常用命令

Docker 包括三个基本概念:镜像（Image）：Docker 镜像（Image），就相当于是一个 root 文件系统。比如官方镜像 ubuntu:16.04 就包含了完整的一套 Ubuntu16.04 最小系统的 root 文件系统。容器（Container）：镜像（Image）和容器（Container）的关系，就像是面向对象程序设计中的类和实例一样，镜像是静态的定义，容器是镜像运行时的实体。容器可以被创建、启动、停止、删除、暂停等。仓库（Repository）：仓库可看成一个代码控制中心

2020-07-08 19:54:27 212

原创 PIL通过exif旋转图像

一些图片是带有exif信息的，某些软件在读取图片时会通过exif信息对图片进行旋转，如果图片是训练集，并且带有标注框，那我们需要以相同的方向显示图片以及对应的标注框假设图像的方向exif标记设置为3。我要做的是根据这个标签旋转这个图像，然后以这种方式保存它。这样，一个不解释exif的软件仍将以良好的方向显示它。但是，如果旋转后保存的图片exif标记方向仍然设置为3，那么解释exif的软件将再次旋转已旋转的图像。所以需要把这个标签设置为1（也就是说：没有方向）或者删除它下面代码是如何通过exif信息判

2020-07-06 14:29:40 1234

空空如也

空空如也