mb_records.append(record) else: raise Exception( "Unimplemented save type: " + str(config.test_save_type)) if config.test_save_type == 'motherboard': #'coco': save_file.write(json.dumps(mb_records)) save_file.close() return save_filename def make_parser(): parser = argparse.ArgumentParser('test network') parser.add_argument( '-d', '--devices', default='0', type=str, help='device for testing') parser.add_argument( '--show_image', '-s', default=False, action='store_true') parser.add_argument('--start_epoch', '-se', default=1, type=int) parser.add_argument('--end_epoch', '-ee', default=-1, type=int) return parser if __name__ == '__main__': parser = make_parser() args = parser.parse_args() args.devices = misc.parse_devices(args.devices) if args.end_epoch == -1: args.end_epoch = args.start_epoch eval_all(args)
coco_records.append(record) else: raise Exception( "Unimplemented save type: " + str(config.test_save_type)) if config.test_save_type == 'coco': save_file.write(json.dumps(coco_records)) save_file.close() return save_filename def make_parser(): parser = argparse.ArgumentParser('test network') parser.add_argument( '-d', '--devices', default='0', type=str, help='device for testing') parser.add_argument( '--show_image', '-s', default=False, action='store_true') parser.add_argument('--start_epoch', '-se', default=1, type=int) parser.add_argument('--end_epoch', '-ee', default=-1, type=int) return parser if __name__ == '__main__': parser = make_parser() args = parser.parse_args() args.devices = misc.parse_devices(args.devices) if args.end_epoch == -1: args.end_epoch = args.start_epoch eval_all(args)