@Author：Runsen

本次博客參考GIthub項(xiàng)目：

• https://github.com/qqwweee/keras-yolo3.git

在開始之前，請(qǐng)?jiān)谝韵挛恢每寺?git clone https://github.com/qqwweee/keras-yolo3.git
到您的本地機(jī)器。確保設(shè)置虛擬環(huán)境并安裝要求

• Keras 2.1.5
• Tensorflow 1.6.0

本次教程分為以下步驟：

• 從 YOLO 網(wǎng)站下載 YOLOv3 權(quán)重。
• 將 Darknet YOLO 模型轉(zhuǎn)換為 Keras 模型。
• 運(yùn)行YOLO檢測(cè)。

代碼環(huán)境：colab

在https://pjreddie.com/darknet/yolo//下載YOLO 模型的權(quán)重。通過運(yùn)行以下命令下載 yolo3 模型的權(quán)重：

%tensorflow_version 1.x !pip3 uninstall keras !pip3 install keras==2.1.5 cd keras-yolo3 !wget https://pjreddie.com/media/files/yolov3.weights

使用convert.py腳本將模型轉(zhuǎn)換為 Keras 兼容文件。

!python convert.py yolov3.cfg yolov3.weights model_data/yolo.h5

在 model_data 目錄中有一個(gè)名為 yolo.h5 的文件。該模型現(xiàn)在可用于通過 yolo_video.py 文件進(jìn)行測(cè)試。

wget https://raw.githubusercontent.com/tensorflow/models/master/research/object_detection/test_images/image1.jpgpython yolo_video.py --model model_data/yolo.h5 --classes model_data/coco_classes.txt --image

如果未指定 --image 標(biāo)志，則需要指定視頻的路徑以及將輸出保存到的路徑。

keras-yolo3 訓(xùn)練自己的自定義 YOLO 模型。需要按照以下步驟操作（來源官方 README）：

下面嘗試使用一個(gè)數(shù)據(jù)集進(jìn)行訓(xùn)練；

• https://www.kaggle.com/tannergi/microcontroller-detection
  

我們需要將csv 變成 yolo的txt

可以通過執(zhí)行voc_annotation.py腳本將 csv 文件轉(zhuǎn)換為 txt 。

文件 model_data/yolo_weights.h5 用于加載預(yù)訓(xùn)練權(quán)重。

對(duì)于微控制器數(shù)據(jù)集，訓(xùn)練命令如下所示：

import xml.etree.ElementTree as ET import argparse import os import globdef voc_to_yolo(input_path, output_path, output_filename, image_folder, classes):# save classesclasses_file = open(os.path.join(output_path, 'classes.txt'), 'w')for c in classes:classes_file.write(c + '\n')classes_file.close() # create txt fileoutput_file = open(os.path.join(output_path, output_filename), 'w')for xml_file in glob.glob(input_path + '/*.xml'):tree = ET.parse(xml_file)root = tree.getroot()image_path = os.path.join(image_folder,root.find("filename").text).replace("\\", "/")output_file.write(f'{image_path} ')for member in root.findall('object'):output_file.write(f'{int(member[4][0].text)},{int(member[4][1].text)},{int(member[4][2].text)},{int(member[4][3].text)},{classes.index(member[0].text)} ')output_file.write('\n')output_file.close()if __name__ == '__main__':parser = argparse.ArgumentParser(description='VOC to YOLO')parser.add_argument('-i', '--input', type=str, required=True, help='Path to input folder')parser.add_argument('-f', '--folder', type=str, default='', help='Image folder (if filenames are relativ)')parser.add_argument('-c', '--classes', nargs='+', required=True, help='Classes')parser.add_argument('-o', '--output', type=str, default='./', help='Output path')parser.add_argument('-of', '--output_filename', type=str, default='train.txt', help='Output Filename')args = parser.parse_args()voc_to_yolo(args.input, args.output, args.output_filename, args.folder, args.classes)python voc_annotation.py -i train/ -f train/ -c Arduino_Nano Heltec_ESP32_Lora ESP8266 Raspberry_Pi_3 -o ./

修改train.py，開始訓(xùn)練。
在該train.py文件中，必須更改annotation_path和classes_path以匹配步驟 1 中創(chuàng)建的文件的路徑。可能還需要更改第 57 行和第 76 行中的 batch_size，具體取決于顯卡具有多少VRAM。

更改 train.py 文件后，您可以通過調(diào)用開始訓(xùn)練：python train.py

train.py一共訓(xùn)練50個(gè)epoch

將開始訓(xùn)練過程并將所有文件保存在指定的日志目錄中。

測(cè)試

python yolo_video.py --model logs/000/trained_weights_final.h5 --classes classes.txt --image

使用yolo_video.py記得修改類路徑或錨路徑。

總結(jié)

以上是生活随笔為你收集整理的深度学习和目标检测系列教程 14-300：训练第一个 YOLOv3 检测器的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。