欢迎大家关注笔者，你的关注是我持续更博的最大动力


原创文章，转载告知，盗版必究


mmdetection2.3.0训练报错：RuntimeError: The size of tensor a （8） must match the size of tensor b （81） at non-singleton dimension 1

文章目录：

1 错误原因分析

1、使用SAC-Cascade模型训练，模型位置：

./configs/detectors/cascade_rcnn_r50_sac_1x_coco.py

2、修改类别数num_classes，在如下的配置文件中修改

./configs/base/models/cascade_rcnn_r50_fpn.py

3、使用多GPU训练命令

./tools/dist_train.sh ./configs/detectors/cascade_rcnn_r50_sac_1x_coco.py 3 --work-dir 23_work_dirs_sac_cascade_hat_hair_beard

然后运行包如下错误，即使使用单GPU依然会报如下错误：
在这里插入图片描述

这个错误原因是由于num_classes没有完全修改导致，

2 解决方式

修改如下配置文件中：num_classes的类别为自己的类别

./configs/base/models/cascade_rcnn_r50_fpn.py

注意：

这个文件中一共有三处 num_classes=80，把这个num_classes全部修改为自己的类别即可

在这里插入图片描述

在这里插入图片描述
♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠

GitHub 加速计划 / mm / mmdetection
28.78 K
9.33 K
下载
open-mmlab/mmdetection: 是一个基于 PyTorch 的人工智能物体检测库，支持多种物体检测算法和工具。该项目提供了一个简单易用的人工智能物体检测库，可以方便地实现物体的检测和识别，同时支持多种物体检测算法和工具。
最近提交(Master分支：1 个月前 )
cfd5d3a9 8 个月前
2390ebc3 8 个月前
Logo
GitCode 开源社区

旨在为数千万中国开发者提供一个无缝且高效的云端环境，以支持学习、使用和贡献开源项目。

更多推荐

[转载]在Windows环境下安装GNU Radio

转自：在Windows环境下安装GNURadio_恐弱智_新浪博客GNU Radio是用Python开发的，大部分开源的工程能够在Linux环境下运行良好，而Windows下却运行的很勉强，而且安装配置都很复杂。GNU Radio算是个例外了，不光提供了Windows的二进制安装，还有比较详细的说明。我是Python小白，所以折腾了好久才弄好，特意记录下来，免得以后再装还折腾。GNU Radio的

avatar GitCode 开源社区

centOS 8 使用dnf安装Docker

DNF是什么？CentOS 8使用YUM软件包管理器版本v4.0.4。现在，该版本使用DNF(已删除YUM)。DNF是软件包管理器。它会在Linux发行版上安装，执行更新并删除软件包。使用DNF安装Docker跳过具有损坏依赖性的程序包一个有效的解决方案是使您的CentOS 8系统使用以下--nobest命令安装最符合条件的版本：sudo dnf install docker...

avatar GitCode 开源社区

定时同步数据库表(mysql+linux+crontab)

sync.sh里面的参数需要改变，ip/username/password/database/tablesync.sh#!/bin/sh# Please change the IP and password of the data source db.# Then change the table name.filename=/home/nington/db/$(date +%Y-%m

avatar GitCode 开源社区