利用faster rcnn进行分类时获取类别名

python版本faster rcnn代码中,使用pascal_voc类型数据进行训练时,pascal_voc初始化时,self._classes是直接赋值的,如self._classes = ('__background__', '2','research'),当类别数较多时,此方法容易遗漏类别号,因此修改为从文件读取的方式。

#coding:utf-8

import os

def obtainclsname(path):
    pclslist = ['__background']
    frcls = open(path,'r')
    for line in frcls:
        line = line.split()
        print ('line is : ', line[0])
        pclslist.append(line[0])
    frcls.close()
    tup = tuple(pclslist)
    return tup
def obtainclsname2(path):
    pclslist = ['__background']
    with open(path,'r') as frcls:
        for line in frcls:
            line = line.split()
            print ('line is : ', line[0])
            pclslist.append(line[0])
    return tuple(pclslist)
if __name__ == '__main__':
    path = 'E:/VOC2007/card/clsname.txt'
    pq = obtainclsname(path)
    print pq
    pqw = obtainclsname2(path)

    print pqw

假设,存储类别的txt文件为:

则在pscal_voc.py中,执行以下操作:


则self._classes为:


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值