def get_test_roidb(imdb_test):
"""Returns a roidb (Region of Interest database) for use in test."""
print('Preparing test data...')
if cfg.TRAIN.HAS_RPN:
rdl_roidb.prepare_roidb(imdb_test)
else:
rdl_roidb.prepare_roidb(imdb_test)
print('done')
return imdb_test.roidb
def get_training_roidb(imdb):
"""Returns a roidb (Region of Interest database) for use in training."""
if cfg.TRAIN.USE_FLIPPED:
print('Appending horizontally-flipped training examples...')
imdb.append_flipped_images()
print('done')
print('Preparing training data...')
rdl_roidb.prepare_roidb(imdb)
print('done')
return imdb.roidb
def get_training_roidb(imdb):
"""Returns a roidb (Region of Interest database) for use in training."""
if cfg.TRAIN.USE_FLIPPED:
print('Appending horizontally-flipped training examples...')
imdb.append_flipped_images()
print('done')
print('Preparing training data...')
if cfg.TRAIN.HAS_RPN:
rdl_roidb.prepare_roidb(imdb)
else:
rdl_roidb.prepare_roidb(imdb)
print('done')
return imdb.roidb
def get_training_roidb(imdb):
"""Returns a roidb (Region of Interest database) for use in training."""
if cfg.TRAIN.USE_FLIPPED:
print 'Appending horizontally-flipped training examples...'
imdb.append_flipped_images()
print 'done'
print 'Preparing training data...'
if cfg.TRAIN.HAS_RPN:
if cfg.IS_MULTISCALE:
gdl_roidb.prepare_roidb(imdb)
else:
rdl_roidb.prepare_roidb(imdb)
else:
rdl_roidb.prepare_roidb(imdb)
print 'done'
return imdb.roidb
def get_training_roidb(imdb):
"""Returns a roidb (Region of Interest database) for use in training."""
print("数据增强之前")
print(len(imdb.roidb))
if cfg.TRAIN.USE_FLIPPED:
print('Appending horizontally-flipped training examples...'
) #这个函数扩增数据集 在imdb 基础上进一步封装
imdb.append_flipped_images()
print('done')
print("数据增强之后") #数据增强通过
print(len(imdb.roidb))
print('Preparing training data...')
rdl_roidb.prepare_roidb(imdb)
#print((imdb.roidb[5523]['image'])) 测试为啥没有image 路径
print('done')
return imdb.roidb
def get_training_roidb(imdb):
"""Returns a roidb (Region of Interest database) for use in training."""
if cfg.ZLRM.TRAIN.USE_FLIPPED:
print('Appending horizontally-flipped training examples...')
imdb.append_flipped_images()
print('done')
print('Preparing training data...')
if cfg.ZLRM.TRAIN.HAS_RPN:
if cfg.ZLRM.IS_MULTISCALE:
# TODO: fix multiscale training (single scale is already a good trade-off)
print('#### warning: multi-scale has not been tested.')
print(
'#### warning: using single scale by setting IS_MULTISCALE: False.'
)
gdl_roidb.prepare_roidb(imdb)
else:
rdl_roidb.prepare_roidb(imdb)
else:
rdl_roidb.prepare_roidb(imdb)
print('done')
return imdb.roidb
