SOLO icon indicating copy to clipboard operation
SOLO copied to clipboard

training my dataset error

Open LiangYong1216 opened this issue 3 years ago • 6 comments

KeyError: 'Pig_Data is not in the dataset registry' when i training my dataset,an error occured,i have registed my dataset,but i don not know why this error occured???

LiangYong1216 avatar Nov 26 '20 08:11 LiangYong1216

@LiangYong1216 Can you post your related code? I can help check it.

WXinlong avatar Nov 30 '20 03:11 WXinlong

those are my related code: My environment:         cuda:10.1  cudnn:7.6.5  mmcv:0.2.16  mmdet:1.0 pytorch:1.4 torchvision:0.5

my dataset only have 3 classes:          so i set num_classes=4, dataset_type = 'CocoToyDataset' data_root = '/home/liangyong/data/coco/'

#this my dataset: from .coco import CocoDataset from .registry import DATASETS

@DATASETS.register_module class CocoToyDataset(CocoDataset):

CLASSES = ['date', 'fig', 'hazelnut']

from .coco import CocoDataset from .coco_toy import CocoToyDataset from .custom import CustomDataset from .dataset_wrappers import ConcatDataset, RepeatDataset from .loader import DistributedGroupSampler, GroupSampler, build_dataloader from .registry import DATASETS from .voc import VOCDataset from .wider_face import WIDERFaceDataset from .xml_style import XMLDataset

all = [ 'CustomDataset', 'XMLDataset', 'CocoDataset', 'VOCDataset', 'CocoToyDataset', 'CityscapesDataset', 'GroupSampler', 'DistributedGroupSampler', 'build_dataloader', 'ConcatDataset', 'RepeatDataset', 'WIDERFaceDataset', 'DATASETS', 'build_dataset' ] when i training my dateset,an error occured:

UserWarning: Default upsampling behavior when mode=bilinear is changed to align_corners=False since 0.4.0. Please specify align_corners=True if the old behavior is desired. See the documentation of nn.Upsample for details. "See the documentation of nn.Upsample for details.".format(mode))

how should i handle this error? thank you very much!!

LiangYong1216 avatar Dec 03 '20 02:12 LiangYong1216

hello! my dataset is little special .some data has no foreground ,and some data has no background. I don't konw if it would work when facing these dataset.thank you .

gcg0210 avatar Jan 13 '21 12:01 gcg0210

KeyError: 'Pig_Data is not in the dataset registry' when i training my dataset,an error occured,i have registed my dataset,but i don not know why this error occured???

those are my related code: My environment:         cuda:10.1  cudnn:7.6.5  mmcv:0.2.16  mmdet:1.0 pytorch:1.4 torchvision:0.5 my dataset only have 3 classes:          so i set num_classes=4, dataset_type = 'CocoToyDataset' data_root = '/home/liangyong/data/coco/' #this my dataset: from .coco import CocoDataset from .registry import DATASETS @DATASETS.register_module class CocoToyDataset(CocoDataset): CLASSES = ['date', 'fig', 'hazelnut'] from .coco import CocoDataset from .coco_toy import CocoToyDataset from .custom import CustomDataset from .dataset_wrappers import ConcatDataset, RepeatDataset from .loader import DistributedGroupSampler, GroupSampler, build_dataloader from .registry import DATASETS from .voc import VOCDataset from .wider_face import WIDERFaceDataset from .xml_style import XMLDataset all = [ 'CustomDataset', 'XMLDataset', 'CocoDataset', 'VOCDataset', 'CocoToyDataset', 'CityscapesDataset', 'GroupSampler', 'DistributedGroupSampler', 'build_dataloader', 'ConcatDataset', 'RepeatDataset', 'WIDERFaceDataset', 'DATASETS', 'build_dataset' ] when i training my dateset,an error occured: UserWarning: Default upsampling behavior when mode=bilinear is changed to align_corners=False since 0.4.0. Please specify align_corners=True if the old behavior is desired. See the documentation of nn.Upsample for details. "See the documentation of nn.Upsample for details.".format(mode)) how should i handle this error? thank you very much!!

i got the same error, how did you solve it ?

jolt2017 avatar Apr 06 '21 09:04 jolt2017

@LiangYong1216 Can you post your related code? I can help check it. 很可能是 mmdet 没有正确安装,我重新安装环境后 问题解决。

jolt2017 avatar Apr 13 '21 02:04 jolt2017

follow the instructions 'Use my own datasets' from https://github.com/WXinlong/SOLO/blob/master/docs/GETTING_STARTED.md#use-my-own-datasets Don't forget to change the 'dataset_type' from 'CocoDataset' to your 'Pig_Data' in config file(eg. configs/solo/*.py)

ssocean avatar May 17 '21 13:05 ssocean