batch_size = nGPUs * config.TRAIN.BATCH_IMAGES print("batch size is", batch_size) if not os.path.isdir(config.output_path): os.mkdir(config.output_path) # Create roidb image_sets = [iset for iset in config.dataset.image_set.split('+')] roidbs = [load_proposal_roidb(config.dataset.dataset, image_set, config.dataset.root_path, config.dataset.dataset_path, proposal=config.dataset.proposal, append_gt=True, flip=config.TRAIN.FLIP, result_path=config.output_path, proposal_path=config.proposal_path, load_mask=config.TRAIN.WITH_MASK, only_gt=not config.TRAIN.USE_NEG_CHIPS) for image_set in image_sets] roidb = merge_roidb(roidbs) roidb = filter_roidb(roidb, config) bbox_means, bbox_stds = add_bbox_regression_targets(roidb, config) print('Creating Iterator with {} Images'.format(len(roidb))) train_iter = MNIteratorE2E(roidb=roidb, config=config, batch_size=batch_size, nGPUs=nGPUs, threads=config.TRAIN.NUM_THREAD, pad_rois_to=400) #, crop_size=(config.TRAIN.SCALES[-1],config.TRAIN.SCALES[-1])) print('The Iterator has {} samples!'.format(len(train_iter))) # Creating the Logger logger, output_path = create_logger(config.output_path, args.cfg, config.dataset.image_set) # get list of fixed parameters print('Initializing the model...')
nGPUs = len(context) batch_size = nGPUs * config.TRAIN.BATCH_IMAGES if not os.path.isdir(config.output_path): os.mkdir(config.output_path) # Create roidb image_sets = [iset for iset in config.dataset.image_set.split('+')] roidbs = [load_proposal_roidb(config.dataset.dataset, image_set, config.dataset.root_path, config.dataset.dataset_path, proposal=config.dataset.proposal, append_gt=True, flip=config.TRAIN.FLIP, result_path=config.output_path, proposal_path=config.proposal_path, load_mask=config.TRAIN.WITH_MASK) for image_set in image_sets] roidb = merge_roidb(roidbs) roidb = filter_roidb(roidb, config) bbox_means, bbox_stds = add_bbox_regression_targets(roidb, config) print('Creating Iterator with {} Images'.format(len(roidb))) train_iter = MNIteratorE2E(roidb=roidb, config=config, batch_size=batch_size, nGPUs=nGPUs, threads=config.TRAIN.NUM_THREAD, pad_rois_to=400) print('The Iterator has {} samples!'.format(len(train_iter))) # Creating the Logger logger, output_path = create_logger(config.output_path, args.cfg, config.dataset.image_set) # get list of fixed parameters print('Initializing the model...')