- 博客(7)
- 资源 (2)
- 收藏
- 关注
原创 Win10系统.pt权重使用TenserRTx转.engine权重文件
简单来说就是使用tensorrt写了一些通用模型的框架,可以将.pt文件转化成.engine文件,供C++调用。修改D:\work_folder\tensorrtx-master\yolov5目录下的yololayer.h文件,设置成自己的预测类别总数。是一个C/C++编程接口,允许程序员在Linux/UNIX下检索关于文件和目录的信息。,根据自己的CUDA下载合适的版本,选择压缩包文件下载。(GA表示稳定版本,EA表示测试版本)将TensorRT目录下lib下的dll文件放到CUDA的bin目录下。
2024-04-14 14:01:14
1123
2
原创 Efficient-Net
模型缩放不会改变在baseline中层的操作,所以有一个好的baseline就十分重要了,我们在现有的卷积神经网络中验证了缩放方法,但是为了更好的展示高效的缩放方法,我们设计了一个新的轻量级baseline,称之为EfficientNet。SENet的提出动机非常简单,传统的方法是将网络的Feature Map等权重的传到下一层,SENet的核心思想在于建模通道之间的相互依赖关系,通过网络的全局损失函数自适应的重新矫正通道之间的特征相应强度。,关于搜索策略是怎么实现的,在论文中的参考文献中。
2024-04-14 13:58:10
603
1
原创 PatchMatch算法详解(python代码实现)
一、简介patchmatch算法的核心目的是在两张图片之间快速寻找对应的小区域。在应用上,patchmatch算法可以结合图像重组等技术,来实现诸如图像修复、图片融合、去水印等功能。patch是指以某个像素为中心的3*3或者5*5的范围。其他的资料很多,建议多搜索一些,本篇重点讲代码是怎么实现的。本篇参考代码来自https://github.com/MingtaoGuo/PatchMatch使用python实现,我的python版本为3.8初学者分享,有错误的地方请指正!二、框架
2021-08-12 10:33:03
9153
9
原创 神经网络八股笔记整理
神经网络八股人工智能实践:Tensorflow笔记 主讲人:曹健来源慕课import导包import tensorflow as tftrain, test切分数据集,训练集数据x_train、训练集标签x_test、测试集数据y_train、测试集标签y_testmodel = tf.keras.models.SequentialSequential函数是一个容器,通过指定参数来生成一个网络层拉直层tf.keras.layers.Flatten().
2021-03-14 13:35:59
319
原创 tensorflowGPU安装+经验分享
历时两天,踩过很多坑,终于语气词装好了。说一下我的情况:tensorflowGPU-1.14.0,CUDA-10.0,cuDNN-v7.6.5,Anaconda3-2019.10,python-3.6,1650显卡。2020年3月2日好了下面是步骤!大致的步骤为一、安装CUDA和cuDNN。二、安装Anaconda三、安装tensorflowGPU...
2020-03-02 23:43:05
2195
2
Java Web程序设计任务教程源码(黑马程序员)
2017-12-11
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人