记录贴：运行swin遇到assert palette.shape[0] == len(self.CLASSES)

最新推荐文章于 2024-05-08 19:59:37 发布

安联之夜

最新推荐文章于 2024-05-08 19:59:37 发布

阅读量553

点赞数 1

文章标签：深度学习

本文链接：https://blog.csdn.net/anlianzhiye/article/details/129464979

版权

一、原因分析

因为用的预训练模型、config文件所用的数据集是Ade20k，而默认的参数palette里的数据集是cityscapes，不一致导致的。
在这里插入图片描述

二、更改

在目录site-packages\mmseg\core\evaluation下有个文件class_name可以找到ade20k在palette里的参数被命名为ade，所以把default的参数改为ade就可以正常运行demo了。

确定要放弃本次机会？

福利倒计时

: :

立减 ¥

普通VIP年卡可用

立即使用

安联之夜

关注关注

1
点赞
踩
1

收藏

觉得还不错? 一键收藏
3
评论
记录贴：运行swin遇到assert palette.shape[0] == len(self.CLASSES)

swinbug记录
复制链接

扫一扫

YOLOv7来临：论文解读附代码解析.pdf

05-20

YOLOv7的主要特点是显著提升了在保持高精度的同时，检测速度也得到了极大提升，相比前几代YOLO模型以及其他的先进检测器如YOLOv3、YOLOv4、YOLOX、Dual-Swin-T、ConvNext和SWIN-L，YOLOv7在速度上有显著优势。...

mx300swin64101ea24.exe

06-12

佳能MX300驱动，佳能MX308驱动，佳能Canon Pixma MX300打印机驱动 WIN7 64位

3 条评论您还未登录，请先登录后发表或查看评论

Python assert 断言函数

qq_41629348的博客

08-05

178

assert 断言的执行逻辑是： if 表达式的值为 True：程序继续运行； else： # 表示式的值为 False 程序引发 AssertionError 错误 assert(b.shape == (5,1)) #b的形状为(5...

语义分割中图片和mask的可视化

周周周周周大帅的博客

05-17

7989

其实取标题一直以来都是一件麻烦的事，但是如果你要看下去，我想你得有一点语义分割的见解。用平常的语言描述该问题就是：语义分割出我们感兴趣的目标物，然后输出该目标物的轮廓点。做语义分割其实有很多种方法，你可以用不同的模型去train你的dataset，但是刚接触语义分割的朋友们可能会说，我该怎么分割出我想要的目标物，而不对其它部分的像素做修改？其实这件事并不复杂，一个很直觉的想法是：修改像素对应的RGB值。那怎么修改对应像素的RGB值呢？举例来说，假设你在mmsegmentation框架下用的是SegFor

解决AssertionError：The length of platte should not be less than ‘num_classes‘

最新发布

cD_nf的博客

05-08

328

主要报错的位置是倒数第三行和第二行代码，意思是dataset_palette的长度小于了目标检测物体的类别num_classes，所以问题出现在dataset_palette的赋值上，分析上面的dataset_palette的赋值过程，当palette == ‘random’或palette is None时，dataset_palette的长度会等于num_classes，所以将倒数第二个报错中传入的palette参数改为None。

mmsegmentation的分割结果可视化

一千零一夜的博客

04-18

4748

mmsegmentation：https://github.com/open-mmlab/mmsegmentation img--->infer,softmax,argmax--->seg def show_result(self, img, result, palette=None, win_name='', .

paddlespeech报错[ ERROR] - list index out of range

qq_37016358的博客

07-28

1987

之后如果有numpy的方法报错，也进行降级到1.23.5后可以正常访问，程序运行过程中出现的警告信息不影响程序正常返回结果。按照官方文档安装paddlepaddle后，使用paddlespeech语音合成，标点恢复功能均正常。由于paddlepaddle是根据官方推荐的安装版本，使用的2.5，引起部分库版本的不兼容。强制降低版本到2.4.2及其他一些依赖降级，就可以跑了，我的python是3.10版本的。降低paddlepaddle后可以正常使用。

Swin-Transformer.zip

12-06

Swin-Transformer图像分类实战，详见文章：https://blog.csdn.net/hhhhhhhhhhwwwwwwwwww/article/details/121744503?spm=1001.2014.3001.5501

Swin Transformer v2实战：使用Swin Transformer v2实现图像分类

10-12

Swin Transformer v2解决了大型视觉模型训练和应用中的三个主要问题，包括训练不稳定性、预训练和微调之间的分辨率差距以及对标记数据的渴望。最新更改：重新适配了timm，并将更换了huggingface的国内链接。链接...

swin.edu.au-esp8266:软件开发项目——esp2866芯片项目

05-30

ESP2866芯片项目“物联网” 软件开发项目 - SWE30008 主管：拉杰什·瓦萨客户：詹姆斯·哈姆林-哈里斯这是一个顶点项目，涉及提供使用 esp8266 芯片的指南。 ESP8266是一款非常便宜（<$ 5美元）的设备，具有...

【飞桨PaddleSpeech语音技术课程】— 声音分类

小湉湉的博客

10-27

2691

【飞桨PaddleSpeech语音技术课程】— 声音分类

OpenPCDet初级教程【自定义模型、loss】

Freak16的博客

08-21

4474

OpenPCDet新手向教程，用小项目的形式记录自定义模块、网络、loss实现训练。

问题记录： assert len(padding_value) == patch.shape[-1]AssertionError

m0_51010465的博客

04-13

134

这里padding value是默认值padding_value=[104, 116, 124]，所以显然长度为3；这里好像(b,w,h,c)应该变为(b,c,h,w)而不是(b,c,w,h)，需要注意。

成功解决[AssertionError: Input tensor input format are different]

u014264373的博客

05-29

6092

在使用tensorboard writer.add_image时 writer.add_image('img/fixed_img', denorm(fixed_img.data), 0) 报如下错误 assert(len(tensor.shape) == len(input_format)), "size of input tensor and input format are different. AssertionError: size of input tensor and input forma

python&opencv视频中人脸的识别

会飞的鱼

04-21

2565

上一篇文章中介绍了静态的人脸的检测，下面介绍动态的即视频中的人脸的检测。原理与上一篇基本相同，只是把视频读取成一帧一帧的图片。 import cv2 face_cascade = cv2.CascadeClassifier('haarcascade_frontalface_alt.xml')#haar人脸识别参考上一篇文章 face_cascade.load('E:\python\ha...

使用超算跑DCF代码时遇到的问题：AssertionError

qq_50988206的博客

10-26

243

这句代码的意思是self.images跟self.gts的长度不相等，才会出现报错，找到self.images和self.gts对应的文件夹，发现两者长度果然不同，是由于自己马虎，导致train_images文件夹里的一部分图片没上传上，上传对应图片，再次运行即不再报错，真是太马虎了。

mmsegmentation日常学习与源码解读

xjw0645的博客

03-13

1164

mmsegmentation保存预测的结果，不覆盖原图首先找到源码中对应的文件 base.py 对应的代码片段如下： def show_result(self, img, result, palette=None, win_name='', show=False, .

【mmdetection】mmdetection推理时，去掉显示结果框的文字标注以及置信度

Gorgeous_wxd

02-16

4415

路径：mmdetection/mmdet/core/visualization/image.py 更改的地方是在imshow_det_bboxes这个函数中更改后的image.py文件 # Copyright (c) OpenMMLab. All rights reserved. import matplotlib.pyplot as plt import mmcv import numpy as np import pycocotools.mask as mask_util from matplotli

def generate(self): if self.backbone not in ['vit_b_16', 'swin_transformer_tiny', 'swin_transformer_small', 'swin_transformer_base']: self.model = get_model_from_name[self.backbone](num_classes=self.num_classes, pretrained=False) else: self.model = get_model_from_name[self.backbone](input_shape=self.input_shape, num_classes=self.num_classes, pretrained=False) device = torch.device('cuda' if torch.cuda.is_available() else 'cpu') self.model.load_state_dict(torch.load(self.model_path, map_location=device)) self.model = self.model.eval() print('{} model, and classes loaded.'.format(self.model_path)) if self.cuda: self.model = nn.DataParallel(self.model) self.model = self.model.cuda()

06-02

4. `self.model = get_model_from_name[self.backbone](input_shape=self.input_shape, num_classes=self.num_classes, pretrained=False)`：调用get_model_from_name函数，通过主干网络名称获取对应的分类模型，并...