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
  • 相关阅读:
    web安全培训笔记
    《virtualbox完全学习手册》
    vim多标签,多窗口
    lnmp.org一键安装包
    git基本命令,Git的skil-map,git配置http/https/socks5代理,,,,,,,,,,,,,,,,,,,,,,
    大数据
    chinacloud大数据新闻
    CentOS6.5/7安装配置Samba
    java项目
    学习Java Web开发
  • 原文地址:https://www.cnblogs.com/aimhabo/p/12857706.html
Copyright © 2011-2022 走看看