物体识别
一只奋进的小蜗牛
这个作者很懒,什么都没留下…
展开
-
TensorFlow模型测试时遇到内存不足问题
2020-11-04 16:55:57.656543: I tensorflow/stream_executor/cuda/cuda_driver.cc:830] failed to allocate 11.00G (11811160064 bytes) from device: CUDA_ERROR_OUT_OF_MEMORY: out of memory2020-11-04 16:55:57.662293: I tensorflow/stream_executor/cuda/cuda_driver.c原创 2020-11-04 17:25:40 · 1803 阅读 · 2 评论 -
Darknet的yolov3.weights文件转换成tensorflow的ckpt或者pb文件
https://blog.csdn.net/heiheiya/article/details/91437196?utm_medium=distribute.pc_relevant.none-task-blog-title-2&spm=1001.2101.3001.4242转载 2020-11-04 14:53:25 · 894 阅读 · 0 评论 -
Tensorflow 2.0之TF-slim
要在TF2中以兼容的模式使用,你需要把它当作一个包来安装。安装方式:Download Zip。然后,python setup.py install使用方法:#import tensorflow as tfimport tensorflow.compat.v1 as tf#from tensorflow.contrib.slim.nets import vggimport tf_slim as slim...原创 2020-11-04 14:32:45 · 2261 阅读 · 0 评论 -
ImportError:No module named builtins
File "darknet2caffe.py", line 10, in <module> from cfg import * File "/home/admin1/darknet2caffe-master/cfg.py", line 1, in <module> import torch File "/home/admin1/.local/lib/python2.7/site-packages/torch/__init__.py", line 19, i..原创 2020-08-04 09:04:21 · 6476 阅读 · 0 评论 -
Numpy问题:ImportError: cannot import name ‘_validate_lengths‘
Traceback (most recent call last): File "darknet2caffe.py", line 7, in <module> import caffe File "/home/admin1/caffe/python/caffe/__init__.py", line 1, in <module> from .pycaffe import Net, SGDSolver, NesterovSolver, AdaGradSolver,原创 2020-08-03 21:37:30 · 750 阅读 · 0 评论 -
numpy.AxisError: axis 2 is out of bounds for array of dimension2
在利用训练好的模型进行物体识别时,输入的图像进行数据转换时遇到问题:接下来讲的是np.expand_dims函数的作用和用法,是用来调整数组维度的。比如有一个二维数组data(1,2),但是你需要输入3维数组,然后你就需要把数据再加一维。但是需要选择加在哪一维。二维数组就有三个位置(0, 1,2)。注意axis的值是从0开始的代码:import numpy as npa = np.array([[[1,2,3],[4,5,6]]])print(a.shape)print('--------原创 2020-08-03 21:18:47 · 14782 阅读 · 2 评论 -
ImportError: No module named torch
ImportError: No module named torchimage and video datasets and models for torch deep learningThe torchvision package consists of popular datasets, model architectures, and common image transformations for computer vision.pip install torchvision原创 2020-08-03 21:08:34 · 1019 阅读 · 0 评论 -
win10+caffe+vs2015+cpu版本搭建过程
第一次在win10系统中安装caffe,一路跌跌撞撞,遇到各种坑,花了1天时间才将环境搭建好,现在记录一下,希望对各位读者也有帮助 关于环境: Visual Studio版本:vs2015,链接:https://pan.baidu.com/s/1xS0XrITmpj6-q5mvmPMv2A 提取码:bt18 Anaconda3:Anaconda3-5.3.1-Windows-x86_64,下载地址:https://mirrors.tuna.tsinghua.edu.cn/a...原创 2020-08-03 20:59:19 · 601 阅读 · 6 评论 -
使用清华源安装opencv-python
直接利用pip install opencv-python 安装速度慢得跟蚂蚁爬一样,还会经常掉线;使用https://www.lfd.uci.edu/~gohlke/pythonlibs/#opencv下载也相当费时间,会掉线;在清华源下载whl文件安装流程比较复杂;所以,使用pip命令,采用清华镜像源安装,快得飞起。pip install opencv-python -i https://pypi.tuna.tsinghua.edu.cn/simple...原创 2020-07-02 14:52:55 · 7160 阅读 · 4 评论 -
Tensorflow2.0学习笔记 tensorboard的使用
ImportError: You need the TensorFlow (v1) module installed to use TensorBoard.解决方法:tf.keras.callbacks.TensorBoard(log_dir=log_dir)就可以了原创 2020-06-18 19:00:37 · 3207 阅读 · 6 评论 -
TensorFlow报错:
TensorFlow报错:AttributeError: module ‘tensorflow_core._api.v2.train’ has no attribute 'Optimizer‘原因:TensorFlow版本更新到2.0,Optimizer方法调用方式改变 。将tf.train.Optimizer,更改为tf.optimizers.Optimizer即可。原创 2020-06-18 18:48:13 · 210 阅读 · 0 评论 -
关于 keras.backend属性错误
解决方案:本人的keras==2.2.51.由于keras版本太高了,可将keras版本降至2.1.5;2.在不改变版本的情况下,将 K.control_flow_ops.while_loop改为tf.while_loop。以上两种方法均可以运行正常。原创 2020-06-18 18:35:27 · 645 阅读 · 2 评论 -
TypeError: Tensor is unhashable if Tensor equality is enabled. Instead, use tensor.experimental_ref(
报错:TypeError: Tensor is unhashable if Tensor equality is enabled. Instead, use tensor.experimental_ref() as the key.`这是tensorflow版本的问题,tensorflow改版后,从V1到V2,很多的东西变化了,导致用V1写的代码,在V2的框架下会报错。这个报错的解决办法:import tensorflow as tftf.compat.v1.disable_v2_behavior(原创 2020-06-18 11:07:38 · 3915 阅读 · 12 评论 -
Tensorfow2.0不能使用get_session问题
RuntimeError: get_session is not available when using TensorFlow 2.0.yolov3中源代码有这样:self.sess = K.get_session()报错显示tf2.0以上移除了这函数,网上大部分方法都是降低tf版本,解决如下:将self.sess = K.get_session()改为self.sess = tf.compat.v1.Session()...原创 2020-06-18 10:30:28 · 3071 阅读 · 3 评论