每天一篇MMSeg
文章平均质量分 74
关于MMSeg自搭建网络学习
weightOneMillion
学海无涯,而我是沧海一粟
展开
-
MMSeg分析Flops和Params
所有data中有数据,同时,input_shape通过默认参数得到,两个不能同时有值,所以将data注释掉。希望通过输入的参数计算。后面可跟参数shape控制输入图片尺寸,例如。这里看一下FlopAnalyzer是如何使用。Flops计算量,params参数量。如何计算某一模块的计算量和参数量呢?在87行左右,由于配置文件在配置了。原创 2024-04-25 15:30:02 · 82 阅读 · 0 评论 -
MMSeg搭建自己的网络
首先,我们知道MMSeg矿机的配置文件很多,主要结构如下图所示。在configs/_base_下是模型配置、数据集配置、以及一些其他的常规配置和运行配置,四类。configs/all_config目录下存放,即是将四种配置聚合在一起的一个总体文件。接下来,我们就搭建自己的网络。原创 2024-04-24 16:54:18 · 637 阅读 · 0 评论 -
MMSeg搭建模型的坑
自己搭建模型的时候,经常会遇到二者不匹配,以这种情况为例,是因为部分模型没有加载到CUDA上面造成的。以下是错误代码,我搞了半天,不理解为什么总是不对,正确的应该是。注意搭建模型的时候,所有层都应该在init函数中完成初始化。搭建代码时,希望记住。没有系统教学,只能自己在网上摸索。其次,对于List、Tuple这种类型,不建议直接用。原创 2024-04-24 15:58:34 · 228 阅读 · 0 评论