This commit is contained in:
YuDeng 2020-12-07 11:36:47 +08:00 коммит произвёл GitHub
Родитель d3d509ae1e
Коммит 9a263d1a7e
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: 4AEE18F83AFDEB23
1 изменённых файлов: 1 добавлений и 2 удалений

Просмотреть файл

@ -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)