# -*- coding: utf-8 -*-
# from util import *
import cv2
import numpy as np
import tensorflow as tf
# from tensorflow.python.framework import graph_util
import os
os.environ['CUDA_DEVICE_ORDER'] = 'PCI_BUS_ID'
os.environ['CUDA_VISIBLE_DEVICES']
TensorFlow直接使用ckpt模型predict不用restore
最新推荐文章于 2024-01-16 09:00:00 发布
该博客介绍了如何在TensorFlow中不通过restore操作,直接使用.ckpt模型文件进行预测。内容涉及检查并理解ckpt模型中的节点名称,以便有效地进行预测流程。
摘要由CSDN通过智能技术生成