Python 为掩码RCNN加载掩码（VGG图像注释器）时出现问题 问题

我想训练一个MRCNN使用预先训练好的椰子重量来检测2个新类别（底部，碗）。我已经在VGG图像注释器中屏蔽了大约400幅图像，我无法加载。我认为我创建的类和多边形有问题，但因为有400多个图像，我真的不想从头开始

基本上我使用的是matterport的代码

错误 在读取带有类和多边形的.json文件时，我遇到错误：

Traceback (most recent call last):
  File ".\Test.py", line 302, in <module>
    train(model)
  File ".\Test.py", line 170, in train
    dataset_train.load_meooow(args.dataset, "train")
  File ".\Test.py", line 115, in load_meooow
    num_ids = [int(n['class']) for n in objects]
  File ".\Test.py", line 115, in <listcomp>
    num_ids = [int(n['class']) for n in objects]
ValueError: invalid literal for int() with base 10: 'bottom'

掩蔽。。。 在上课期间，我有时会改变上课顺序。我的意思是，有时班底是一班，碗是二班

1 bottom
2 bowl

1 bowl
2 bottom

有时碗是一级的，底部是二级的

1 bottom
2 bowl

1 bowl
2 bottom

当有什么东西盖在班级碗上时，我不得不把它分成两个多边形，但名称相同。所以我在VGG中有1个底部，2个碗，3个碗

1 bottom
2 bowl
3 bowl

这可能是个问题，或者只要我坚持使用正确的多边形标签，我就可以了

标签

---

我不能使用像rcnn、mrcnn、mask rcnn这样的标签，因为它至少需要1500个声誉。。抱歉…

如果您不想将类名注释为整数字符串“1”，可以像我在本例中所做的那样尝试以下代码：

classes_name = ['bottom', 'bowl']
num_ids = [classes_name.index(n['class']) + 1 for n in objects]

---

这个错误是因为在VGG中添加了类。在我的例子中，在注释时，我使用类名代替类号。 删除类名并用类号替换它。在你的情况下，替换 在VGG中注释时，底部为1，碗为2