0

Tensorflow框架中GPU的使用与否不是只会影响速度吗,为什么还会影响运行结果。。

关闭GPU后得到的结果会是一片空白。。。使用GPU则正常

https://pan.baidu.com/s/1mfA7...

代码如上,可以直接运行的

主要代码如下:

# -*- 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)
Shark 1
2018-08-26 提问
0 个回答

撰写答案

推广链接