dragonfly91-CSDN博客

原创 Kafka连接报错：kafka.errors.NoBrokersAvailable: NoBrokersAvailable

最近做的项目，搭建的kafka服务器要通过外网进行生产和消费，但是发现不管是生产还是消费都会卡住，消息发不出去，但是topic可以正常查看，kafka服务也能初始化。最后发现需要修改server.properties中的一行，然后重启kafka和zookeeper服务即可。修改如下#listeners=PLAINTEXT://:9092advertised.listeners=PLAINTEXT://x.x.x.x:9092x.x.x.x代表服务器对外的IP...

2021-11-18 09:56:34 4598 1

原创 pycuda._driver.LogicError: cuDeviceGet failed: initialization error解决方法

在使用Python API进行TensorRT推理时遇到了这个错误，出错的代码行，形如：# Create a Context on this device,self.cfx = cuda.Device(0).make_context()报错为：pycuda._driver.LogicError: cuDeviceGet failed: initialization error后来谷歌了一下，发现是import的时候少了一项，只要加上import pycuda.autoinit问题解

2021-05-17 11:07:26 3589 3

原创 pth-＞onnx-＞trt踩坑笔记1

1、最近编译一个torch的pth转TensorRT的trt工具的时候遇到的错误：/usr/bin/ld: 找不到 -lnvinfer/usr/bin/ld: 找不到 -lcudartcollect2: error: ld returned 1 exit statusmake[2]: *** [CMakeFiles/dbnet.dir/build.make:120：dbnet] 错误 1make[1]: *** [CMakeFiles/Makefile2:95：CMakeFiles/db

2021-01-26 17:31:30 1919 3

原创 Tensorrt7.2.1安装问题libcudnn.so.8: cannot open shared object file: No such file or directory

最近安装TensorRT7.2.1，cuda版本10.2，cudnn8.02，按照官网教程安装完成之后，import tensorrt的时候遇到问题。ImportError: libcudnn.so.8: cannot open shared object file: No such file or directory我的cuda是做了一个软链接，软链接地址在/usr/local/cuda下，我首先将/usr/local/cuda/lib64的路径，添加到.bashrc里面，即export LD

2020-12-18 15:04:53 8120 3

原创 Pascal Voc格式转ICDAR2015格式的脚本

最近做OCR的项目，需要将Pascal Voc的格式转换为ICDAR2015的格式，所以写了这个转换脚本。importosimportxml.etree.ElementTreeasETxmldir='Annotations'gtdir='ICDAR_gt'ifnotos.path.exists(gtdir):os.mkdir(gtdir)xmlfiles=os.listdir(xmldir)forxmlfileinxmlfiles:...

2020-11-23 10:52:40 396 1

原创 Leetcode 221. Maximal Square Python3

题目描述：Given a 2D binary matrix filled with 0's and 1's, find the largest square containing only 1's and return its area.Example:Input:1 0 1 0 01 0 1 1 11 1 1 1 11 0 0 1 0Output: 4动态规划方法...

2020-02-26 23:03:35 140

原创 Leetcode 31. Next Permutation Python3

算法推导如何得到这样的排列顺序？这是本文的重点。我们可以这样来分析：我们希望下一个数比当前数大，这样才满足“下一个排列”的定义。因此只需要将后面的大数与前面的小数交换，就能得到一个更大的数。比如 123456，将 5 和 6 交换就能得到一个更大的数 123465。我们还希望下一个数增加的幅度尽可能的小，这样才满足“下一个排列与当前排列紧邻“的要求。为了满足这个要求，我们需要：在尽可能靠...

2020-02-21 23:34:04 148

原创 Leetcode 343. Integer Break 整数拆分（剪绳子）Python3 & C++

此题使用动态规划的思路来做。将数字i分割为i-j和j，一种备选是计算j*(i-j)，另一种备选是继续分割i-j，由于i-j肯定比i小，所以可以根据动态规划的思想由前面的值得到。问题的起点，i=1的时候最大就为1。Python3代码：class Solution: def max3(self,a,b,c): return max(max(a,b),c)...

2020-02-12 17:18:28 173

原创牛客网剑指offer 数组中的逆序对 Python3

思路参考了另一篇博客，此处只是将原作者的代码翻译为Python版本：https://blog.csdn.net/lym940928/article/details/91354887# -*- coding:utf-8 -*-class Solution: def InversePairs(self, data): length=len(data) ...

2020-02-07 16:49:19 153

原创 Leetcode 112. Path Sum 二叉树中和为某一值的路径 Python3

方法：递归最直接的方法就是利用递归，遍历整棵树：如果当前节点不是叶子，对它的所有孩子节点，递归调用 hasPathSum 函数，其中 sum 值减去当前节点的权值；如果当前节点是叶子，检查 sum 值是否为 0，也就是是否找到了给定的目标和。# Definition for a binary tree node.# class TreeNode:# def __init...

2020-02-04 21:56:05 172

原创牛客网剑指offer 二叉搜索树的后序遍历序列 Python3

题目要求：输入一个整数数组，判断该数组是不是某二叉搜索树的后序遍历的结果。如果是则输出Yes,否则输出No。假设输入的数组的任意两个数字都互不相同。采用分治法的思想，找到根结点、左子树的序列、右子树的序列，分别判断左右子序列是否为二叉树的后序序列。由题意可得：1. 后序遍历序列的最后一个元素为二叉树的根节点；2. 二叉搜索树左子树上所有的结点均小于根结点、右子树所有的结点均大于根...

2020-02-04 17:49:21 159

原创 Leetcode 19. Remove Nth Node From End of List 链表中倒数第n个结点Python3版本

方法：两次遍历算法思路我们注意到这个问题可以容易地简化成另一个问题：删除从列表开头数起的第 (L - n + 1)个结点，其中 L是列表的长度。只要我们找到列表的长度 L，这个问题就很容易解决。算法首先我们将添加一个哑结点作为辅助，该结点位于列表头部。哑结点用来简化某些极端情况，例如列表中只含有一个结点，或需要删除列表的头部。在第一次遍历中，我们找出列表的长度 L。然后设置一个指向哑...

2020-02-03 17:25:29 137

原创 Leetcode 232. Implement Queue using Stacks用两个栈实现队列

使用最经典的两个栈的方式实现队列，下面是python3的实现方式class MyQueue: def __init__(self): """ Initialize your data structure here. """ self.stack1 = [] self.stack2 = [] d...

2020-02-02 19:09:17 109

原创 leetcode题解 426. Convert Binary Search Tree to Sorted Doubly Linked List

方法一：递归算法标准的中序遍历采用左 -> 根 -> 右的顺序，其中左和右的部分调用递归。本题的处理在于将前一个结点与当前结点链接，因此，必须跟踪最后一个结点，该结点在新的双向链表中是当前最大的。另外一个细节：我们也需要保留第一个，也就是最小的结点，以完成闭环。下面是具体算法：将 first 和 last 结点初始化为 null。调用标准中序遍历...

2020-01-26 21:28:51 302

原创 Pytorch模型转换，pth->onnx->trt(TensorRT engine)

pth->onnx常见问题##模型输入输出不支持字典在使用torch.onnx导出onnx格式时，模型的输入和输出都不支持字典类型的数据结构。**解决方法：**此时，可以将字典数据结构换为torch.onnx支持的列表或者元组。例如：heads {'hm': 1, 'wh': 2, 'hps': 34, 'reg': 2, 'hm_hp': 17, 'hp_offset'...

2019-11-28 18:41:48 9970 4

原创 DCNv2编译找不到nvcc问题

在编译DCNv2 0.1版本的时候，遇到nvcc库找不到的问题，因为系统默认的nvcc库路径是/usr/bin/nvcc，此时需要手动设置相关的LD_LIBRARY_PATH。我的.bashrc的修改如下：export PATH=/usr/local/cuda/bin:$PATHexport LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_...

2019-11-21 12:24:55 1523 1

原创解决ubuntu安装TensorFlow1.14.0版本时ERROR: Cannot uninstall 'wrapt'问题

报错信息如下：Cannot uninstall 'wrapt'. It is a distutils installed project and thus we cannot accurately determine which files belong to it which would lead to only a partial uninstall.此时，输入：pip insta...

2019-07-18 10:44:18 2678 1

原创 206. Reverse Linked List Python迭代实现

题目要求：Reverse a singly linked list.Example:Input: 1->2->3->4->5->NULLOutput: 5->4->3->2->1->NULLFollow up:A linked list can be reversed either iteratively or ...

2019-05-05 23:04:50 234

转载 U-Net论文详解

U-Net：生物医学图像分割的卷积神经网络U-net 是基于FCN的一个语义分割网络,适合用来做医学图像的分割。摘要有许多成功利用大量带标注训练数据的神经网络。在这篇论文里，我们提出一个网络和训练策略，更有效的利用了数据增强来使用可获得的带标注数据。这个算法包括一个收缩路径来捕捉语义和一个扩展路径来精准定位。这个网络可以端到端训练非常少的图片，但是在ISBI（电子显微镜的细胞图像分...

2019-04-21 20:50:02 706

原创 Boltztrap安装和使用简述

1、安装python2、安装pip，python库管理器3、利用pip库管理器，安装spglib和ase两个库4、库安装好之后，调好.bashrc，然后开始安装boltztrap5、编译boltztrap，直接make，按照readme编译即可，编译完成后，将其路径添加到.bashrc里面6、利用vasp静态计算后的文件，要有EIGENVAL,OUTCAR,POSCAR7、

2017-11-17 19:25:02 11616 1

dragonfly91的博客