2017-11-23 13 views
0

ディープ・ラーニングにFaster RCNN libraryを使用しています。ここでは、自分のデータセットを訓練する方法についてdiscussionです。それが私より先の一歩です。トレーニングのためのRCNNライブラリの読み込みトレーニングデータセットはどれくらいですか?

トレーニングのためのデータセットがどのようにセットアップされ、トレーニング用にどのようにロードされているかを理解したいですか?私はコードを見てみると

、私はtrain_faster_rcnn_alt_opt.pyからコード

imdb = get_imdb(imdb_name)のこのラインを見て、それがfactory.py inside datasets folderを呼び出します。

factory.pyは、私はこのIMDB名voc_2007_trainvalのためのトレーニングデータはトレーニングのためにロードされているか疑問に思って

for year in ['2007', '2012']: 
    for split in ['train', 'val', 'trainval', 'test']: 
     name = 'voc_{}_{}'.format(year, split) 
     __sets[name] = (lambda split=split, year=year: pascal_voc(split, year)) 

# Set up coco_2014_<split> 
for year in ['2014']: 
    for split in ['train', 'val', 'minival', 'valminusminival']: 
     name = 'coco_{}_{}'.format(year, split) 
     __sets[name] = (lambda split=split, year=year: coco(split, year)) 

# Set up coco_2015_<split> 
for year in ['2015']: 
    for split in ['test', 'test-dev']: 
     name = 'coco_{}_{}'.format(year, split) 
     __sets[name] = (lambda split=split, year=year: coco(split, year)) 

def get_imdb(name): 
    """Get an imdb (image database) by name.""" 
    if not __sets.has_key(name): 
     raise KeyError('Unknown dataset: {}'.format(name)) 
    return __sets[name]() 

def list_imdbs(): 
    """List all registered imdbs.""" 
    return __sets.keys() 

を持っていますか?

編集: __sets[name]() inside def get_imdb(name):を印刷すると、次のように表示されます。

p __sets[name]() 
<datasets.pascal_voc.pascal_voc object at 0x7fc937383ed0> 

これはどういう意味ですか?

+0

'pascal_voc'クラスの' datasets/pascal_voc.py'を見てください。 – Shai

+1

@ Tanks Shai、私は方法を見つけた。 – batuman

答えて

0

今私は理解しました。 のlib /データセット/ factory.pyたちは訓練コマンドで--imdb voc_2007_trainvalを設定しているため、どのようなことがないことは、プログラムが内部で述べた画像を読み込むことができるということです

for year in ['2007', '2012']: 
    for split in ['train', 'val', 'trainval', 'test']: 
     name = 'voc_{}_{}'.format(year, split) 
     __sets[name] = (lambda split=split, year=year: pascal_voc(split, year)) 

呼び出す

def get_imdb(name): 
    """Get an imdb (image database) by name.""" 
    if not __sets.has_key(name): 
     raise KeyError('Unknown dataset: {}'.format(name)) 
    return __sets[name]() 

__sets [名前]を持っていますtrainvalファイルdata/VOCdevkit2007/VOC2007/ImageSets/Main/trainval.txtファイルです。

--imdb voc_2007_trainを設定すると、train.txtが使用されます。すべての画像はJPEGImagesフォルダ内にあり、注釈はAnnotationsフォルダにあります。

関連する問題