任務(wù)目標(biāo)
1. 根據(jù)給定的網(wǎng)絡(luò)模型代碼與數(shù)據(jù)集,修改Yolov10模型的檢測頭,使其與給定模型的檢測頭輸出一致
2. 給出更換后Yolov10模型和給定模型的性能對比,包括:準(zhǔn)確率,召回率,檢測速度
3. 完成模型在orin平臺的部署及優(yōu)化
交付需求
1. 開發(fā)環(huán)境
1. conda環(huán)境
2. 遷移后的網(wǎng)絡(luò)模型:torch格式,onnx格式
3. 模型訓(xùn)練代碼
1. 數(shù)據(jù)預(yù)處理:格式適配,數(shù)據(jù)增強(qiáng)等
2. 訓(xùn)練參數(shù)配置
3. 損失函數(shù)配置
4. 訓(xùn)練超參數(shù)調(diào)優(yōu)方案
4. 驗(yàn)證與測試代碼:
1. 模型測試代碼
2. 測試集評估結(jié)果
5. 模型導(dǎo)出代碼:導(dǎo)出成onnx格式
6. 完成Orin芯片環(huán)境部署:
1. 設(shè)置部署環(huán)境,確保依賴項(xiàng)合硬件支持
2. 部署配置與測試
7. 詳細(xì)文檔
1. 模型結(jié)果,訓(xùn)練過程,部署步驟
2. 模型測試報告
數(shù)據(jù)說明
開發(fā)數(shù)據(jù)由我們提供,為通過魚眼相機(jī)轉(zhuǎn)換得到的線性圖片