Verbesserung von YOLOv8 | Upgrade des Backbone-Netzwerks | YOLOv8-Backbone-Netzwerk ersetzt durch MobileNetV3 | Computer Vision
YOLOv8 ist derzeit einer der beliebtesten Objektdetektoren und seine neueste Version bietet schnellere und genauere Erkennungsfunktionen. Die Geschwindigkeit und Genauigkeit von YOLOv8 hängen stark von seinem Backbone-Netzwerk ab. In diesem Artikel wird beschrieben, wie Sie die Leistung von YOLOv8 verbessern können, indem Sie das Backbone-Netzwerk durch MobileNetV3 ersetzen.
MobileNetV3 ist ein leichtes Faltungs-Neuronales Netzwerk, das 2019 von Google vorgeschlagen wurde. Im Vergleich zu anderen Backbone-Netzwerken bietet MobileNetV3 eine schnellere Berechnungsgeschwindigkeit und eine höhere Genauigkeit. Wir werden MobileNetV3 verwenden, um die Verarbeitungsgeschwindigkeit von YOLOv8 zu beschleunigen.
Zuerst müssen wir die vorab trainierten Gewichte von MobileNetV3 herunterladen. Dieses Gewicht kann aus mehreren Quellen bezogen werden. Hier entscheiden wir uns dafür, es von TensorFlow Hub zu beziehen.
import tensorflow as tf
import tensorflow_hub as hub
# 下载 MobileNetV3 预训练权重
module_url = "https://tfhub.dev/google/imagenet/mobilenet_v3_large_100_224/classification/5"
model = tf.keras