![](https://img-blog.csdnimg.cn/20201014180756925.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
目标识别
文章平均质量分 63
默语..
努力就是给青春最好的答案!
展开
-
pytorch环境下进行模型测试总是报OOM(out of memory)的解决办法
当模型训练好以后,进行测试或者线上部署模型时候,时长会遇到OOM(out of memory)的问题;那么,是什么原因导致的呢?原创 2021-07-13 19:07:15 · 2500 阅读 · 0 评论 -
深度学习:使用Tensorflow Lite部署模型时遇到不支持tf.keras.layers.UpSampling2D()函数问题的解决方法
在使用Tensorflow Lite部署训练好的模型时,若使用了tf.keras.layers.UpSampling2D()函数对图像进行了上采样,那么就会报错——Tensorflow Lite暂不支持该函数方法。那么该如何解决呢,可以使用tf.image.resize()方法来代替UpSampling2D方法。为验证该方案是否可行,下面简单的进行了一下验证:import tensorflow as tfimport numpy as np# 自定义一张图片矩阵input_image = tf.原创 2021-04-20 17:43:53 · 824 阅读 · 0 评论 -
Ubuntu20.04(Deepin20.1)环境下安装显卡驱动、cuda/cudnn及源码编译opencv-4.5.1详细教程
目录一、安装前准备二、显卡驱动安装三、cuda安装四、cudnn安装五、opencv-4.5.1源码编译参考一、安装前准备显卡驱动下载:https://www.nvidia.cn/Download/index.aspx?lang=cn创建文件:sudo vim /etc/modprobe.d/blacklist-nouveau.conf在文件中添加以下内容,将nouveau加入黑名单:blacklist nouveauoptions nouveau modeset=0启用黑名单原创 2021-03-14 08:27:35 · 1013 阅读 · 1 评论 -
深度学习:利用Detectron2训练自己的数据集之模型评估——解决AP=0问题
本文主要阐述模型评估相关内容,遇到的坑及解决办法在进行模型评估的时候遇到有些自定义的类别的AP=0。经过原创 2021-03-10 21:55:49 · 4376 阅读 · 16 评论