weixin_42912072的博客

AssertionError: The num_classes (3) in Shared2FCBBoxHead of MMDataParallel does not matches the …一届书生# 2021-10-13 11:07:00 85 收藏 1文章标签： 深度学习 tensorflow tomcat anaconda python版权mmdetection训练自己的数据集时报错 ⚠️ ：AssertionError: The num_classes (3) in Shared2

环境ubuntu18.04 TLpython1.6pytorch 1.0.0torchision0.2.2ubuntu环境下我使用conda创建的虚拟环境，cudatoolkit10.1.243（就是大家所说的CUDA）。cuda驱动是450版本，这个没有太大的影响只要他能支持起来cudatoolkit 10.1.243就可以。以下是2条指令：conda create -n CenterNet python=3.6pip install torch==1.0.0 torchvision

训练单类别的时候一定要和其他的类别一起训练，即使没有那也得写上几个。如果你选的class_numbers==1的话就等着报错吧，keyname error好像是这个。所以记住了哦。

http://tzutalin.github.io/labelImg/

在mmdetection的安装说明中https://github.com/open-mmlab/mmdetection/blob/master/docs/install.md。直接操作两行代码即可，首先clone到主代码git clone https://github.com/open-mmlab/mmdetection.git然后cd进入mmd环境cd mmdetection运行setup安装文件python setup.py develop...

此次的cifar10和前面说的MNIST案例虽然主骨架是相同的，但是代码内部有很大的区别，相同点：他们都是采用了2层卷积+2层全连接不同点：cifar10内部封装了数据增强的功能，而且在全连接层cifar10应用了L2正则项来约束w参数，防止过拟合，并没有采用MNIST的那种dropout，代码如下：import tensorflow as tfimport numpy as npimpo...

一、卷积神经网络卷积神经网络首先从他的起源和原理来了解和熟悉下他，下面咱们来看看百度百科摘得的一些资料。卷积神经网络（Convolutional Neural Network,CNN）是一种前馈神经网络，它的人工神经元可以响应一部分覆盖范围内的周围单元，对于大型图像处理有出色表现。 [2] 它包括卷积层(convolutional layer)和池化层(pooling layer)。20世...

下面咱们用tf自带的MNIST数据集来实现一下手写体数字识别import tensorflow as tffrom tensorflow.examples.tutorials.mnist import input_data#加载数据并转换mnist = input_data.read_data_sets(“MNIST_data_bak/”, one_hot=True)sess = tf....