这个算法怎么训练自己的数据集啊,在哪更改数据集的类别数和类别?
cfg.MODEL.POSITION_HEAD.THING.NUM_CLASSES = 80
cfg.MODEL.POSITION_HEAD.STUFF.NUM_CLASSES = 54
cfg.MODEL.SEM_SEG_HEAD.NUM_CLASSES = 54
这三个类别classes分别是啥意思?这个80是coco数据集的类别数,但是这个类别名称在哪更改?
关于平台上的项目PanopticFCN实例分割算法
这个算法怎么训练自己的数据集啊,在哪更改数据集的类别数和类别?
cfg.MODEL.POSITION_HEAD.THING.NUM_CLASSES = 80
cfg.MODEL.POSITION_HEAD.STUFF.NUM_CLASSES = 54
cfg.MODEL.SEM_SEG_HEAD.NUM_CLASSES = 54
这三个类别classes分别是啥意思?这个80是coco数据集的类别数,但是这个类别名称在哪更改?
你好,这里面所有关于seg有关的都是语义分割的类别,而thing是实例类别。也就是检测里面的类别。
而语义是不区分实例的,在coco里面实例类别是80类,而语义是54类。
如果需要修改就需要修改成你的类别数,加入你的数据存放格式和coco一样,那么这些就只要修改数字即可,如果不是,你得实现自己的dataloader。