-
Notifications
You must be signed in to change notification settings - Fork 0
/
generate_multi.py
33 lines (26 loc) · 1.28 KB
/
generate_multi.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
import argparse
from core.models.gas_model import GASNet_MV
import core.data.voxelized_data_shapenet as voxelized_data
from core.generator import Generator
from core.generation_iterator import gen_iterator
parser = argparse.ArgumentParser(description='Run generation')
parser.add_argument('-res' , default=64, type=int)
parser.add_argument('-retrieval_res' , default=128, type=int)
parser.add_argument('-checkpoint', type=int)
parser.add_argument('-cat','--category' , default='airplane', type=str)#'chair' or 'airplane'
parser.add_argument('-n', '--num_views', default=3, type=int)# 2 or 3
args = parser.parse_args()
net = GASNet_MV(args.res, args.category)
exp_name = '{}v{}_{}'.format('GAS_MV', args.res, args.category)
dataset = voxelized_data.VoxelizedDataset('test',
args.category,
batch_size=1,
num_workers=0)
gen = Generator(net,
0.5,
exp_name,
checkpoint=args.checkpoint,
resolution=args.retrieval_res,
num_views=args.num_views)
out_path = 'experiments/{}/evaluation_{}@{}_{}v/'.format(exp_name, args.checkpoint, args.retrieval_res, args.num_views)
gen_iterator(out_path, dataset, gen)