问题解决
小恩子
这个作者很懒,什么都没留下…
展开
-
Python3.7之后,sklearn 的linear_assignment不支持,用scipy代替
import numpy as npfrom sklearn.utils.linear_assignment_ import linear_assignmentfrom scipy.optimize import linear_sum_assignment as linear_sum_assignmentconst_matrix=np.array([[15,40,45],[20,60,35],[20,40,25]])matches=linear_assignment(const_matrix).原创 2021-11-30 18:16:51 · 2493 阅读 · 4 评论 -
1.fatal error: NvInfer.h: No such file or directory
编译yolov5生成engine,报/tensorrtx/yolov5/yololayer.h:6:10: fatal error: NvInfer.h: No such file or directory6 | #include <NvInfer.h> | ^~~~~~~~~~~compilation terminated.1、未找到tensorRT对应的头文件和库。在CMakeLists.txt#tensorrt处添加tensorrt对应头文件和...原创 2021-11-22 09:25:30 · 3902 阅读 · 0 评论 -
ubuntu升级python和pip
由于各种原因需要升级python到更高版本,同时需要建立pip和python对应映射。推荐两个博客:1、在Ubuntu 16.04(18.04) 系统上编译安装Python 3.8 方法2、ubuntu16.04纯净版-安装Python3.8.1/升级pip原创 2021-11-21 22:19:16 · 1050 阅读 · 0 评论 -
TX2 pytorch 安装
TX2 jetpack4.5 对应cuda10.2,需要与pytorch对应,不然没法使用,版本对应如下。TX2 torch-1.7.0.0-cp36-cp36m1-linux_aarch64.whl 百度网盘下载:链接:https://pan.baidu.com/s/1lI1opYAdryJQBrNsZLn63w 提取码:ovsg ...原创 2021-06-23 10:49:57 · 312 阅读 · 0 评论 -
Pulsar 订阅存在bug
1、最近在用pulsar开发项目,在项目中不断关闭、开启线程,每一次开启线程都要订阅一个摄像头的topic。发现出现内存泄漏。import pulsar,timeclient = pulsar.Client("pulsar://15.18.162.112:6650")consumer =client.subscribe("non-persistent://public/default/topic-" + "15010200001310004341**", 'smoke-subscription')原创 2021-06-22 16:54:42 · 453 阅读 · 0 评论 -
JETSON TX2 刷机注意事项
最近手上有一块JETSON TX2,网上有很多详细教程就不具体了,但有的教程有误导性。1、host machine可以不勾选,交叉编译使用的,我就没有勾选,jiandi原创 2021-06-16 14:49:08 · 167 阅读 · 0 评论 -
please determine the input size manually by “--dynamic-input-shape --input-shape xxx“ or “--input-sh
最近在做onnx转trt使用onnxsim出现这个错误,源代码如下:from onnxsim import simplifyimport onnxinput_path="cnn.onnx"output_path="sim_cnn.onnx"onnx_model = onnx.load(input_path) # load onnx model)model_simp, check = simplify(onnx_model)assert check, "Simplified ONNX mo原创 2021-06-05 11:44:37 · 1865 阅读 · 0 评论 -
[TensorRT] ERROR: Network must have at least one output[TensorRT] ERROR: Network validation failed.
1、pytorch模型转为onnx后用tensorRT量化加速时,报这个问题。OnnxParser在解析onnx文件时没解析成功,这个问题一方面可能涉及到pytorch和tensorrt版本的问题,另一方面可能涉及到网络模型有些操作, tensorrt中的onnx解析器不支持,解析不了.第一方面, tensorrt6.0系列只能使用pytorch1.2及以下版本. 所以需要使用torch1.2生成onnx文件.第二方面, 就要将一些不支持的操作进行替换, 比如expand_as等操作有时候上原创 2021-06-04 11:21:12 · 1360 阅读 · 2 评论 -
device CUDA:2 not supported by XLA service while setting up XLA_GPU_JIT device number 2
torch运行的时候,每一块显卡都会占用一点显存,如果只想让其运行在指定的显卡上,可以使用下面的命令:export CUDA_VISIBLE_DEVICES=1,2在执行torch程序的时候,先在终端中执行上面的命令,就可以让该终端只看到指定的显卡,上面的命令中的1和2就是nvidia-smi命令看到的显卡的序号.执行了上面的命令后,torch指定的GPU从0开始按照上面可见的依次往后....原创 2021-04-30 11:32:23 · 2372 阅读 · 0 评论 -
RuntimeError: unable to write to file </torch_2818186_3541370554> ERROR: Unexpected bus error encoun
在容器中,训练yolov5时,报如上问题。pytorch将共享内存的临时文件保存在了/torch_xxx文件中,即容器中的根目录下。容器磁盘空间不足导致该问题的发生。目前可以通过以下代码暂时关闭pytorch的shared memory功能来规避:将此代码贴到你启动代码的最前面,顺利解决:import sysimport torchfrom torch.utils.data import dataloaderfrom torch.multiprocessing import reductio原创 2021-04-26 11:23:36 · 1420 阅读 · 0 评论 -
AttributeError: ‘list‘ object has no attribute ‘dtype‘
AttributeError: 'list' object has no attribute 'dtype' tensor_info_output = tf.saved_model.utils.build_tensor_info(model.output)改为 tensor_info_output = tf.saved_model.utils.build_tensor_info(model.output[0])原创 2020-11-11 18:22:11 · 3930 阅读 · 0 评论 -
TypeError: __init__() got an unexpected keyword argument ‘ragged‘
TypeError: __init__() got an unexpected keyword argument 'ragged'from keras.models import load_model 改为from tensorflow.keras.models import load_model原创 2020-11-11 18:00:30 · 1384 阅读 · 0 评论 -
使用imutils包的VideoStream读取视频流,丢帧率达20%
最近在读取相机RTSP流,发现OpenCV不如下面方式读取的快:import _threadfrom imutils.video import VideoStreamimport imutilsimport timeimport numpy as npimport cv2# # 为线程定义一个函数def cam1( threadName): cap = VideoStream('test.mp4').start() n=0 while True:原创 2020-07-14 15:42:13 · 5324 阅读 · 3 评论 -
fatal error: Python.h: 没有那个文件或目录 compilation terminated.
pythonn 3.5 遇上如上问题,网友给的解决方案:sudo apt-get install python3-dev未解决:sudo apt-get install python-dev解决!!!原创 2020-06-30 21:13:12 · 1669 阅读 · 0 评论 -
ModuleNotFoundError: No module named numpy.random._pickle
原因:numpy版本较老,版本不匹配。解决:pip install numpy --upgrade --user原创 2020-06-10 11:26:58 · 7070 阅读 · 2 评论 -
TypeError: namedtuple() got an unexpected keyword argument -verbose
File "C:\Users\Desktop\Online-Realtime-Action-Recognition-based-on-OpenPose-master\Pose\pose_estimator.py", line 21, in <module> 'coord1', 'coord2', 'score1', 'score2'], verbose=False)TypeError: namedtuple() got an unexpected keyword argument 'v..原创 2020-06-03 09:06:41 · 3241 阅读 · 0 评论 -
graph_def.ParseFromString(f.read()) google.protobuf.message.DecodeError: Error parsing message
当测试https://github.com/ildoonet/tf-pose-estimation/ 仓库执行如下命令时python run.py --model=cmu --resize=656x368 --image=./images/p1.jpg遇到如上问题,在网上没有找到解决方案,仔细分析原因:在下载模型时发现下载的模型不全只有几百KB大小;通过如下命令很难把模型完整下载下来。$ cd models/graph/cmu$ bash download.sh模型下载链接:链接: .原创 2020-05-30 19:47:52 · 5645 阅读 · 0 评论