2021 山东大学软件学院软件工程应用与实践--YOLOV5项目代码分析——（8）

最新推荐文章于 2024-07-12 16:16:27 发布

qq_57299978

最新推荐文章于 2024-07-12 16:16:27 发布

阅读量489

点赞数

分类专栏： YOLOV5 文章标签： python 深度学习 pytorch

本文链接：https://blog.csdn.net/qq_57299978/article/details/122164059

版权

2021SC@SDUSC

最后分析tf.py,是基于TensorFlow, Keras and TFLite实现的YOLOv5算法

tf.py

导入相应模块

"""
TensorFlow, Keras and TFLite versions of YOLOv5
Authored by https://github.com/zldrobit in PR https://github.com/ultralytics/yolov5/pull/1127

Usage:
    $ python models/tf.py --weights yolov5s.pt

Export:
    $ python path/to/export.py --weights yolov5s.pt --include saved_model pb tflite tfjs
"""

import argparse
import logging
import sys
from copy import deepcopy
from pathlib import Path

FILE = Path(__file__).resolve()
ROOT = FILE.parents[1]  # YOLOv5 root directory
if str(ROOT) not in sys.path:
    sys.path.append(str(ROOT))  # add ROOT to PATH

import numpy as np
import tensorflow as tf
import torch
import torch.nn as nn
from tensorflow import keras

from models.common import Conv, Bottleneck, SPP, DWConv, Focus, BottleneckCSP, Concat, autopad, C3
from models.experimental import CrossConv, MixConv2d, attempt_load
from m