ModuleNotFoundError: No module named 'tensorflow.python.saved_model.model_utils'

本文介绍了解决TensorFlow1.13.x环境下ModuleNotFoundError问题的方法。通过调整tensorflow-estimator版本和使用PyCharm进行精确版本控制来解决'tensorflow.python.saved_model.model_utils'模块未找到的问题。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

ModuleNotFoundError: No module named 'tensorflow.python.saved_model.model_utils'


网上的方法主要是更新tensorflow-estimator版本,但是值得注意的是,tensorflow13.1要求tensorflow-estimator版本在13.0到14.0之间。
而网上大部分都是因为安装了tensorflow2.0,所以使用下面的方法能够将tensorflow-estimator版本更新到14.0。故可以解决tensorflow2.0出现标题中错误。

 pip uninstall tensorflow_estimator
 pip install tensorflow_estimator

但是当我的tensorflow13.1使用上面的方法,错误不会消除,这里提供一个解决办法:
我使用的是pycharm,在file->setting->project:project_name中可以看到所有安装包的版本
在这里插入图片描述
然后双击版本号这个地方(图中1.13.0)
就可以进入下图,进入后勾选右侧的Specify version就可以选择想要的版本了:
在这里插入图片描述
最后点击左下角install Package就可以了。实测可以解决tensorflow13.1报错的问题。

import os import torch import torch.nn as nn import torch.optim as optim from torch.utils.data import DataLoader from config import * from utils import YOLODataset, get_data_loader, YOLOLoss # 假设已有相关函数 from models.yolov5 import YOLOv5 # 加载数据 train_loader = get_data_loader(TRAIN_IMAGES, TRAIN_LABELS, img_size=IMG_SIZE, batch_size=BATCH_SIZE, shuffle=True) val_loader = get_data_loader(VAL_IMAGES, VAL_LABELS, img_size=IMG_SIZE, batch_size=BATCH_SIZE, shuffle=False) # 初始化模型 model = YOLOv5(num_classes=80).to(DEVICE) criterion = YOLOLoss(num_classes=80, img_size=IMG_SIZE) optimizer = optim.AdamW(model.parameters(), lr=LEARNING_RATE, weight_decay=WEIGHT_DECAY) # 模型保存路径 MODEL_SAVE_PATH = "D:\\commodity_sorting_system\\final_model.pth" # 训练 for epoch in range(EPOCHS): model.train() total_loss = 0 for batch_idx, (images, labels) in enumerate(train_loader): images = images.to(DEVICE) labels = labels.to(DEVICE) optimizer.zero_grad() outputs = model(images) loss = criterion(outputs, labels) loss.backward() optimizer.step() total_loss += loss.item() print(f"Epoch [{epoch+1}/{EPOCHS}], Batch [{batch_idx+1}/{len(train_loader)}], Loss: {loss.item():.4f}") # 训练完成后保存模型 torch.save(model.state_dict(), MODEL_SAVE_PATH) print(f"Training complete. Final model saved to {MODEL_SAVE_PATH}") 运行后显示C:\Users\23228\PyCharmMiscProject\.venv\Scripts\python.exe D:\commodity_sorting_system\code\main.py Traceback (most recent call last): File "D:\commodity_sorting_system\code\main.py", line 8, in <module> from models.yolov5 import YOLOv5 ModuleNotFoundError: No module named 'models.yolov5' 进程已结束,退出代码为 1 怎么解决
06-05
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值