MASK_RCNN训练自己的数据集

最新推荐文章于 2024-02-06 17:19:03 发布

原创最新推荐文章于 2024-02-06 17:19:03 发布 · 1.4k 阅读

4 ·

CC 4.0 BY-SA版权

本文详细介绍了如何使用Mask_RCNN框架训练自定义数据集，包括数据集转换为COCO格式、配置文件调整及常见问题解决。特别强调了数据准备的重要性，以及在训练过程中可能遇到的数据格式和Keras版本问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

MASK_RCNN训练自己的数据集

**
1.将自己数据集的标注转为coco数据集格式，网上有很多很好的代码。这一步比较重要，后期出的很多问题都是数据的问题
在这里插入图片描述
train和val中是原始图片，annotations是放的coco格式的标记信息

2 coco.py 的修改
（1）89行附件

NUM_CLASSES = 1 + 1  # COCO has 80 classes

自己的数据分成几类，就是1+几，这个1是背景

（2）114行附近

		coco = COCO("{}/annotations/{}{}.json".format(dataset_dir, subset, year))
		#由于我是train.json，val.json,所以我把instances删掉了
        if subset == "minival" or subset == "valminusminival":
            subset = "val"
        image_dir = "{}/{}{}".format(dataset_dir, subset, year)

(3)423行附近

  if args.command == "train":
        # Training dataset. Use the training set and 35K from the
        # validation set, as as in the Mask RCNN paper.
        dataset_train = CocoDataset()
        dataset_train.load_coco(args.dataset, "train", year='', auto_download=args.download)
        dataset_train.load_coco(args.dataset, "val", year='', auto_download=args.download)
        dataset_train.prepare()

3.运行coco.py
在cmd的coco.py所在的文件夹下

 python coco.py train --dataset=G:\data\parking\drivable_area\data\ --model=imagenet

还有

# Train a new model starting from pre-trained COCO weights
python3 coco.py train --dataset=/path/to/coco/ --model=coco

# Train a new model starting from ImageNet weights
python3 coco.py train --dataset=/path/to/coco/ --model=imagenet

# Continue training a model that you had trained earlier
python3 coco.py train --dataset=/path/to/coco/ --model=/path/to/weights.h5

# Continue training the last model you trained
python3 coco.py train --dataset=/path/to/coco/ --model=last

# Run COCO evaluatoin on the last model you trained
python3 coco.py evaluate --dataset=/path/to/coco/ --model=last

我的G:\data\parking\drivable_area\data\下：

在这里插入图片描述