Yolos Tiny Hard Hat Detection

🚀 yolos-tiny-Hard_Hat_Detection

該模型是 hustvl/yolos-tiny 在安全帽檢測數據集上的微調版本，可用於物體檢測任務，展示了利用技術解決複雜問題的能力。

🚀 快速開始

此模型是 hustvl/yolos-tiny 在安全帽檢測數據集上的微調版本。若想了解其創建方式的更多信息，請查看以下鏈接：https://github.com/DunnBC22/Vision_Audio_and_Multimodal_Projects/blob/main/Computer%20Vision/Object%20Detection/Hard%20Hat%20Detection/Hard_Hat_Object_Detection_YOLOS.ipynb

✨ 主要特性

模型描述

若想了解該模型的創建方式，請查看此鏈接：https://github.com/DunnBC22/Vision_Audio_and_Multimodal_Projects/blob/main/Computer%20Vision/Object%20Detection/Hard%20Hat%20Detection/Hard_Hat_Object_Detection_YOLOS.ipynb

預期用途與侷限性

此模型旨在展示使用技術解決複雜問題的能力。

訓練和評估數據

數據集來源：https://huggingface.co/datasets/keremberke/hard-hat-detection

訓練過程

訓練超參數

訓練期間使用了以下超參數：

• 學習率（learning_rate）：5e - 05
• 訓練批次大小（train_batch_size）：8
• 評估批次大小（eval_batch_size）：8
• 隨機種子（seed）：42
• 優化器（optimizer）：Adam，β值為(0.9, 0.999)，ε值為1e - 08
• 學習率調度器類型（lr_scheduler_type）：線性
• 訓練輪數（num_epochs）：8

訓練結果

框架版本

• Transformers 4.31.0
• Pytorch 2.0.1+cu118
• Datasets 2.14.3
• Tokenizers 0.13.3

📄 許可證

本模型採用 Apache 2.0 許可證。