Tensorflow框架中GPU的使用与否不是只会影响速度吗,为什么还会影响运行结果。。
关闭GPU后得到的结果会是一片空白。。。使用GPU则正常
代码如上,可以直接运行的
主要代码如下:
# -*- coding: utf-8 -*-
import numpy as np
from scipy.misc import imread, imsave
import argparse
import time
import os
# os.environ["CUDA_VISIBLE_DEVICES"] = "-1" # 参数为 -1 表示只使用 CPU ,注释后可启用GPU加速
import tensorflow as tf
def run(session):
args = argparse.Namespace()
args.input = './7.jpg'
args.output = './7_666.jpg'
args.model = './models/wave.model'
args.arch = "./models/model.meta"
args.image = imread(args.input, mode="RGB").astype(np.float32)
args.image = np.expand_dims(args.image, axis=0)
# 使用模型
saver = tf.train.import_meta_graph(args.arch, clear_devices=True)
saver.restore(session, args.model)
inputs = tf.get_collection("inputs")[0]
output = tf.get_collection("output")[0]
result = output.eval({inputs: args.image})
result = np.clip(result, 0.0, 255.0).astype(np.uint8)
result = np.squeeze(result, 0)
imsave(args.output, result)
def main():
session = tf.Session()
with session.as_default():
run(session)
if __name__ == "__main__":
time_a = time.time()
main()
time_b = time.time()
print('run_time', time_b - time_a)