Update demo.py
This commit is contained in:
Родитель
d3d509ae1e
Коммит
9a263d1a7e
3
demo.py
3
demo.py
|
@ -60,9 +60,8 @@ def demo():
|
|||
# build reconstruction model
|
||||
with tf.Graph().as_default() as graph,tf.device('/cpu:0'):
|
||||
|
||||
opt = Option()
|
||||
opt = Option(is_train=False)
|
||||
opt.batch_size = 1
|
||||
opt.is_train = False
|
||||
opt.pretrain_weights = args.pretrain_weights
|
||||
FaceReconstructor = Face3D()
|
||||
images = tf.placeholder(name = 'input_imgs', shape = [opt.batch_size,224,224,3], dtype = tf.float32)
|
||||
|
|
Загрузка…
Ссылка в новой задаче