zoukankan      html  css  js  c++  java
  • COCO数据集使用super categories时出现IndexError: list index out of range

    在文件xxx/site-packages/pycocotools-2.0-py3.7-linux-x86_64.egg/pycocotools/coco.py中找到约157行左右的函数

    def getCatIds(self, catNms=[], supNms=[], catIds=[]):

    改为:

        def getCatIds(self, catNms=[], supNms=[], catIds=[]):
            """
            filtering parameters. default skips that filter.
            :param catNms (str array)  : get cats for given cat names
            :param supNms (str array)  : get cats for given supercategory names
            :param catIds (int array)  : get cats for given cat ids
            :return: ids (int array)   : integer array of cat ids
            """
            #catNms = catNms if _isArrayLike(catNms) else [catNms]
            #supNms = supNms if _isArrayLike(supNms) else [supNms]
            #catIds = catIds if _isArrayLike(catIds) else [catIds]
    
            #if len(catNms) == len(supNms) == len(catIds) == 0:
            #    cats = self.dataset['categories']
            #else:
            #    cats = self.dataset['categories']
            #    cats = cats if len(catNms) == 0 else [cat for cat in cats if cat['name']          in catNms]
            #    cats = cats if len(supNms) == 0 else [cat for cat in cats if cat['supercategory'] in supNms]
            #    cats = cats if len(catIds) == 0 else [cat for cat in cats if cat['id']            in catIds]
            #ids = [cat['id'] for cat in cats]
            ids = [cat['id'] for cat in self.dataset['categories']]
            #print(self.dataset['categories'])
            #print(cats)
            return ids
  • 相关阅读:
    python-pytest学习(四)-fixture简介
    python-pytest学习(三)-setup/teardown
    python-pytest学习(二)-执行用例规则
    Vue 之五 生命周期钩子函数 自定义指令 过滤器
    Vue 之四 Swiper
    Vue 之三局部组件与全局组件
    Vue 之三 与后端交互
    有关于mysql面试的小故事
    Vue 基础之二
    kubernetes
  • 原文地址:https://www.cnblogs.com/aimhabo/p/12857706.html
Copyright © 2011-2022 走看看