![](https://img-blog.csdnimg.cn/20201014180756913.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
深度学习
SXANG
要做一阵风
展开
-
根据RGB图像中颜色值生成标签灰度图,
1.前言:代码目的:将RGB图像中不同的颜色生成标签灰度图,cyan[青蓝色]映射为0,red[红色]映射为1,yellow[黄色]映射为22.代码# -*- coding: utf-8 -*-"""Created on Sat May 7 16:54:02 2022@author: hpDescribe:将RGB图像中颜色,生成标签灰度图 例:RGB图像中cyan、red、yellow三种不同的颜色区域,需要根据颜色生成灰度图,作为标签数据原创 2022-05-11 21:38:47 · 670 阅读 · 0 评论 -
深度学习中修改了模型结构后,将官方的预训练权重文件转化为自己模型的权重文件
0.前言:在深度学习中,官方提供的模型预训练权重文件对于我们的训练有相当大的作用。因为这些权重文件都是官方使用大型数据集在高性能计算机上训练得到的,我们自己很难在大型数据集上训练出自己模型的预训练权重文件,再迁移到自己的数据集上,因此转化官方的权重文件为自己模型的权重文件很有必要。1.生成自己模型权重文件:这个就很简单啦!在进行模型训练时,训练一轮就能得到模型随机初始化的模型权重文件。这里取名my_weigths.pth2.模型转化:...原创 2022-03-28 15:09:57 · 10571 阅读 · 15 评论