关于tensorflow2.8.0版本训练facenet人脸识别input resource[0] expected type resource != float问题解决

在用tensorflow2.8.0版本进行facenet人脸识别时,采用freeze_graph对模型参数进行固化后,加载pb文件出现以下报错:
在这里插入图片描述报错原因为tensorflow2.版本和tensorflow1.x版本不兼容导致,解决办法为在对train_tripletloss.py模型进行训练时,在导入包的部分加入tf.compat.v1.disable_v2_behavior(),对2.x版本进行屏蔽。

from __future__ import absolute_import
from __future__ import division
from __future__ import print_function

from datetime import datetime
import os.path
import time
import sys
import tensorflow.compat.v1 as tf
import numpy as np
import importlib
import itertools
import argparse
import facenet
import lfw
from tensorflow.python.ops import data_flow_ops
from six.moves import xrange  # @UnresolvedImport
tf.disable_v2_behavior()  # 屏蔽2.x版本兼容1.x版本

训练完成后,对模型固化的时候,同样需要加入tf.compat.v1.disable_v2_behavior(),进行pb文件转化。
参考链接:.https://blog.csdn.net/weixin_45080073/article/details/117034111/

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值