Pytorch-YOLOv3运行中可能遇到的错误

原创于 2019-10-11 11:55:12 发布 · 1.5k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#Pytorch-YOLOv3

yolo 专栏收录该内容

16 篇文章

订阅专栏

本文详细介绍了在使用TensorFlow 2.x版本时遇到的兼容性问题，特别是在PyTorch-YOLOv3项目中。主要问题是由于API更改导致的AttributeError。文章提供了具体的解决方案，即将tf.summary.FileWriter替换为tf.summary.create_file_writer，以确保代码在新版本的TensorFlow中正确运行。

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

github链接：https://github.com/eriklindernoren/PyTorch-YOLOv3

错误1

File "pytorch_platform/PyTorch-YOLOv3/utils/logger.py", line 7, in __init__
    self.writer = tf.summary.FileWriter(log_dir)
AttributeError: module 'tensorboard.summary._tf.summary' has no attribute 'FileWriter'

出现上述错误的原因是我使用的tensorflow版本是2.0版本，其中一些api有改动，解决方法如下：

将    self.writer = tf.summary.FileWriter(log_dir)

改为  self.writer = tf.summary.create_file_writer(log_dir)