Google Colab範例,使用Yolov4-tiny(darknet)對自定義資料集進行訓練及推論
yolov4-tiny_training_test.ipynb 為主程式,包含Yolov4-tiny預訓練測試、自定義資料集訓練及推論測試
my_dataset.zip 包含100張影像及Yolo格式標註檔(*.txt)
my_obj.data 為資料集設定檔
my_obj.names 為資料集分類標籤名稱
my_yolov4-tiny-custom.cfg 為自定義模型及相關參數
my_train.txt 為自定義資料訓練集檔案名稱列表
my_val.txt 為自定義資料驗證集檔案名稱列表
test01.jpg, test02.jpg 為測試用影像
訓練時間約1~1.5小時,視分配到的GPU類型及網路速度。