We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
这里贴一下划分val的代码
coco = COCO(os.path.join(root, 'annotations/instances_train2017.json')) class_names = [coco.cats[catId]['name'] for catId in coco.getCatIds()] categories = [dict(id=i+1, name=name) for i, name in enumerate(class_names)] annotaions_train = [] images_train = [] annotaions_val = [] images_val = [] for catId in coco.getCatIds(): imgIds = coco.getImgIds(catIds=[catId]) random.shuffle(imgIds) for imgId in imgIds[:10]: img = coco.imgs[imgId] images_val.append(img) anns = coco.imgToAnns[imgId] for ann in anns: annotaions_val.append(ann) for imgId in imgIds[10:]: img = coco.imgs[imgId] images_train.append(img) anns = coco.imgToAnns[imgId] for ann in anns: annotaions_train.append(ann) json_dict_train = {"images": images_train, "type": "instances", "annotations": annotaions_train, "categories": categories} json_dict_val = {"images": images_val, "type": "instances", "annotations": annotaions_val, "categories": categories}
json_dict_train, json_dict_val 存储一下就可以了
The text was updated successfully, but these errors were encountered:
No branches or pull requests
这里贴一下划分val的代码
json_dict_train, json_dict_val 存储一下就可以了
The text was updated successfully, but these errors were encountered: