Python set_proposal_methodの例

コード例 #1

ファイル: trainval_net.py プロジェクト: yzhang123/stairs_detection

 def get_roidb(imdb_name):
     imdb = get_imdb(imdb_name)
     print('Loaded dataset `{:s}` for training'.format(imdb.name))
     imdb.set_proposal_method(cfg.TRAIN.PROPOSAL_METHOD)
     print('Set proposal method: {:s}'.format(cfg.TRAIN.PROPOSAL_METHOD))
     roidb = get_training_roidb(imdb)
     return roidb

コード例 #2

 def get_roidb(imdb_name):
     # 获取voc 原始数据集
     imdb = get_imdb(imdb_name)
     print('Loaded dataset `{:s}` for training'.format(imdb.name))
     imdb.set_proposal_method(cfg.TRAIN.PROPOSAL_METHOD)
     # print('Set proposal method: {:s}'.format(cfg.TRAIN.PROPOSAL_METHOD))
     # 制作并获取目标区域的位置信息
     roidb = get_training_roidb(imdb)
     return roidb

コード例 #3

ファイル: trainval_net.py プロジェクト: codecookinging/fasterrcnn_tensorflow-impl

def get_roidb(imdb_name):  #本函数根据数据库名字 获取roidb

    imdb = get_imdb(imdb_name)  #调用factory 的 imdb 然后根据 数据库 名字 调用 pascal_voc
    print('Loaded dataset `{:s}` for training'.format(imdb.name))
    imdb.set_proposal_method(cfg.TRAIN.PROPOSAL_METHOD)  #设置gt 的方法
    print('Set proposal method: {:s}'.format(cfg.TRAIN.PROPOSAL_METHOD))
    roidb = get_training_roidb(imdb)

    return imdb, roidb

コード例 #4

ファイル: train.py プロジェクト: bhfs9999/FasterRCNN_GPU

def get_imdb_roidb(imdb_name, image_type, data_type='cervix'):
    imdb = get_cervex_db(imdb_name, image_type, data_type=data_type)
    print('Loaded dataset `{:s}` for training'.format(imdb.name))
    imdb.set_proposal_method(cfg.TRAIN.PROPOSAL_METHOD)
    roidb = get_training_roidb(imdb)
    return imdb, roidb