- html - 出于某种原因,IE8 对我的 Sass 文件中继承的 html5 CSS 不友好?
- JMeter 在响应断言中使用 span 标签的问题
- html - 在 :hover and :active? 上具有不同效果的 CSS 动画
- html - 相对于居中的 html 内容固定的 CSS 重复背景?
我使用带有 8 个 Titan X 的服务器,试图比使用单个 GPU 更快地预测图像。
我像这样加载 PB 模型:
model_dir = "./model"
model = "nasnet_large_v1.pb"
model_path = os.path.join(model_dir, model)
model_graph = tf.Graph()
with model_graph.as_default():
with tf.gfile.GFile(model_path, 'rb') as f:
graph_def = tf.GraphDef()
graph_def.ParseFromString(f.read())
_ = tf.import_graph_def(graph_def, name='')
input_layer = model_graph.get_tensor_by_name("input:0")
output_layer = model_graph.get_tensor_by_name('final_layer/predictions:0')
./data_input
中的文件目录如下:
with tf.Session(graph = model_graph, config=config) as inference_session:
# Initialize session
initializer = np.zeros([1, 331, 331, 3])
print("Initialing session...")
inference_session.run(output_layer, feed_dict={input_layer: initializer})
print("Done initialing.")
# Prediction
file_list = []
processed_files = []
for path, dir, files in os.walk('./model_output/processed_files'):
for file in files:
processed_files.append(file.split('_')[0]+'.tfrecord')
print("Processed files: ")
for f in processed_files:
print('\t', f)
while True:
for path, dir, files in os.walk("./data_input"):
for file in files:
if file == '.DS_Store': continue
if file in processed_files: continue
print("Reading file {}".format(file))
file_path = os.path.join('./data_input', file)
file_list.append(file_path)
res = predict(file_path)
processed_files.append(file)
with open('./model_output/processed_files/{}_{}_processed_files.json'.format(file.split('.')[0], model.split('.')[0]), 'w') as f:
f.write(json.dumps(processed_files))
with open('./model_output/classify_result/{}_{}_classify_result.json'.format(file.split('.')[0], model.split('.')[0]), 'w') as f:
f.write(json.dumps(res, indent=4, separators=(',',':')))
time.sleep(1)
predict()
函数,我写了这样的代码:
label_map = get_label()
# read tfrecord file by tf.data
dataset = get_dataset(filename)
# dataset.apply(tf.contrib.data.prefetch_to_device("/gpu:0"))
# load data
iterator = dataset.make_one_shot_iterator()
features = iterator.get_next()
result = []
content = {}
count = 0
# session
with tf.Session() as sess:
tf.global_variables_initializer()
t1 = time.time()
try:
while True:
[_image, _label, _filepath] = sess.run(fetches=features)
_image = np.asarray([_image])
_image = _image.reshape(-1, 331, 331, 3)
predictions = inference_session.run(output_layer, feed_dict={input_layer: _image})
predictions = np.squeeze(predictions)
# res = []
for i, pred in enumerate(predictions):
count += 1
overall_result = np.argmax(pred)
predict_result = label_map[overall_result].split(":")[-1]
if predict_result == 'unknown': continue
content['prob'] = str(np.max(pred))
content['label'] = predict_result
content['filepath'] = str(_filepath[i], encoding='utf-8')
result.append(content)
except tf.errors.OutOfRangeError:
t2 = time.time()
print("{} images processed, average time: {}s".format(count, (t2-t1)/count))
return result
with tf.device('/gpu:{}'.format(i))
在加载模型部分或推理 session 部分或 session 部分,
nvidia-smi
显示只有 GPU0 被 100% 使用,而其他 GPU 似乎即使在加载内存时也无法工作。
最佳答案
您可以通过以下方式强制图中每个节点的设备:
def load_network(graph, i):
od_graph_def = tf.GraphDef()
with tf.gfile.GFile(graph, 'rb') as fid:
serialized_graph = fid.read()
od_graph_def.ParseFromString(serialized_graph)
for node in od_graph_def.node:
node.device = '/gpu:{}'.format(i) if i >= 0 else '/cpu:0'
return {"od_graph_def": od_graph_def}
关于python-3.x - 在 TensorFlow 中使用多个 GPU 来推断 pb 模型,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/54247097/
我需要将文本放在 中在一个 Div 中,在另一个 Div 中,在另一个 Div 中。所以这是它的样子: #document Change PIN
奇怪的事情发生了。 我有一个基本的 html 代码。 html,头部, body 。(因为我收到了一些反对票,这里是完整的代码) 这是我的CSS: html { backgroun
我正在尝试将 Assets 中的一组图像加载到 UICollectionview 中存在的 ImageView 中,但每当我运行应用程序时它都会显示错误。而且也没有显示图像。 我在ViewDidLoa
我需要根据带参数的 perl 脚本的输出更改一些环境变量。在 tcsh 中,我可以使用别名命令来评估 perl 脚本的输出。 tcsh: alias setsdk 'eval `/localhome/
我使用 Windows 身份验证创建了一个新的 Blazor(服务器端)应用程序,并使用 IIS Express 运行它。它将显示一条消息“Hello Domain\User!”来自右上方的以下 Ra
这是我的方法 void login(Event event);我想知道 Kotlin 中应该如何 最佳答案 在 Kotlin 中通配符运算符是 * 。它指示编译器它是未知的,但一旦知道,就不会有其他类
看下面的代码 for story in book if story.title.length < 140 - var story
我正在尝试用 C 语言学习字符串处理。我写了一个程序,它存储了一些音乐轨道,并帮助用户检查他/她想到的歌曲是否存在于存储的轨道中。这是通过要求用户输入一串字符来完成的。然后程序使用 strstr()
我正在学习 sscanf 并遇到如下格式字符串: sscanf("%[^:]:%[^*=]%*[*=]%n",a,b,&c); 我理解 %[^:] 部分意味着扫描直到遇到 ':' 并将其分配给 a。:
def char_check(x,y): if (str(x) in y or x.find(y) > -1) or (str(y) in x or y.find(x) > -1):
我有一种情况,我想将文本文件中的现有行包含到一个新 block 中。 line 1 line 2 line in block line 3 line 4 应该变成 line 1 line 2 line
我有一个新项目,我正在尝试设置 Django 调试工具栏。首先,我尝试了快速设置,它只涉及将 'debug_toolbar' 添加到我的已安装应用程序列表中。有了这个,当我转到我的根 URL 时,调试
在 Matlab 中,如果我有一个函数 f,例如签名是 f(a,b,c),我可以创建一个只有一个变量 b 的函数,它将使用固定的 a=a1 和 c=c1 调用 f: g = @(b) f(a1, b,
我不明白为什么 ForEach 中的元素之间有多余的垂直间距在 VStack 里面在 ScrollView 里面使用 GeometryReader 时渲染自定义水平分隔线。 Scrol
我想知道,是否有关于何时使用 session 和 cookie 的指南或最佳实践? 什么应该和什么不应该存储在其中?谢谢! 最佳答案 这些文档很好地了解了 session cookie 的安全问题以及
我在 scipy/numpy 中有一个 Nx3 矩阵,我想用它制作一个 3 维条形图,其中 X 轴和 Y 轴由矩阵的第一列和第二列的值、高度确定每个条形的 是矩阵中的第三列,条形的数量由 N 确定。
假设我用两种不同的方式初始化信号量 sem_init(&randomsem,0,1) sem_init(&randomsem,0,0) 现在, sem_wait(&randomsem) 在这两种情况下
我怀疑该值如何存储在“WORD”中,因为 PStr 包含实际输出。? 既然Pstr中存储的是小写到大写的字母,那么在printf中如何将其给出为“WORD”。有人可以吗?解释一下? #include
我有一个 3x3 数组: var my_array = [[0,1,2], [3,4,5], [6,7,8]]; 并想获得它的第一个 2
我意识到您可以使用如下方式轻松检查焦点: var hasFocus = true; $(window).blur(function(){ hasFocus = false; }); $(win
我是一名优秀的程序员,十分优秀!