- 博客(8)
- 资源 (1)
- 收藏
- 关注
原创 测试Tensorflow、Pytorch的GPU是否可用
测试Tensorflow、Pytorch的GPU是否可用原始环境中为cuda10.0 的Tensorflow1.13.1和Pytorch1.12版本,CUDA通用。虚拟环境中安装了Pytorch1.16,CUDA10.2版本。问题来了:Tensorflow目前最新版本2.4.1竟然不支持CUDA10.2版本,仅支持到CUDA10.1。等着以后支持CUDA10.2在更新版本吧,先用CPU。1.Pytorch测试GPU是否可用2.Tensorflow测试GPU是否可用注: 报了一堆
2021-01-23 12:15:40 706
转载 轻量化网络结构——MobileNet系列
MobileNet系列总结1、概述 MobileNet系列网络是Google为了移动端和嵌入式设备设计的模型,主要是在追求准确性和运行速度之间的平衡。从模型结构上主要对以下几点做了研究:V1:深度可分离卷积V2:倒置残差块 和 线性瓶颈层V3:网络架构搜索V1和V2都是先提出了一种理论、假说、发现,然后根据这种理论来指导模型的设计,这是有一定启发意义的。V3主要是在用NAS搜索结构,使用V2+SENet+swish的基本模块构建网络,设计思路上并没有很大的启发。2、MobileNetV
2021-01-16 16:19:28 1161 2
原创 轻量化网络结构——MobileNetV3
论文:《Searching for MobileNetV3》概述: 该论文主要讲如何使用NAS搜索出适用于移动设备的轻型网络,提出了MobileNetV3-Large和MobileNetV3-Small两个模型。基础结构: 网络结构图如下所示,主要是在点向卷积后加了一个SENet的attention模块。激活函数: 另外用上了swish激活函数。给出了两种形式。最终块的优化:MobileNetV3的网络结构: MobileNetV3定义了两个模型: MobileNetV
2021-01-16 16:04:07 2427
原创 轻量化网络结构——MobileNetV2
《Inverted Residuals and Linear Bottlenecks: Mobile Networks for Classification, Detection and Segmentation》—Google1.MobileNetV2结构是基于倒置残差结构,原来的残差结构的主分支是有三个卷积,两个逐点卷积通道数较多,而倒置的残差结构刚好相反,中间的卷积通道数(依旧使用深度可分离卷积结构)较多,两端的较少。另外,我们发现去除主分支中的非线性变换是有效的,这可以保持模型的表现力。2.论文
2021-01-16 15:33:44 6221 1
原创 轻量化网络结构——Xception
《Xception: Deep Learning with Depthwise Separable Convolutions》– Google1.Xception 并不是真正意义上的轻量化模型,是Google继Inception后提出的对Inception v3的另一种改进,主要是采用depthwise separable convolution来替代原来的Inception v3中的卷积操作,这种性能的提升是来自于更有效的使用模型参数而不是提高容量。一个卷积层尝试去学习特征在3维空间–(高、宽、
2021-01-16 15:16:57 3013
原创 轻量化网络结构——ShuffleNet
论文:《ShuffleNet: An Extremely Efficient Convolutional Neural Network for Mobile Devices》—Face++1.shuffle具体来说是channel shuffle,是将各部分的feature map的channel进行有序的打乱,构成新的feature map,以解决group convolution带来的[信息流通不畅]的问题。(MobileNet是用pointwise convolution解决这个问题。)因此可知
2021-01-16 14:58:27 2017
原创 sklearn报错:joblib
报错原因:原始版本的scikit-learn == 0.19.2,新版本为scikit-learn == 0.23.2。sklearn.externals.joblib函数是用在0.21及以前的版本中,在最新的版本中,该函数应被弃用。所以引入from sklearn.externals import joblib库时,出现ImportError: cannot import name 'joblib’错误。解决方法:高版本直接import joblib 导入即可低版本预测代码:from skl
2021-01-09 12:07:05 4401 1
原创 OMP: Error #15: Initializing libiomp5md.dll, but found libiomp5md.dll already initialized.
运行torch模型时,报错信息如下:OMP: Error #15: Initializing libiomp5md.dll, but found libiomp5md.dll already initialized.OMP: Hint This means that multiple copies of the OpenMP runtime have been linked into the program. That is dangerous, since it can degrade perform
2021-01-06 16:50:12 1293
Windows 环境下编译OpenCV3.4.1和OpenCV-Contrib3.4.1下配置文件
2018-12-16
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人