任務(wù)目標(biāo)
1. 根據(jù)給定的網(wǎng)絡(luò)模型代碼與數(shù)據(jù)集,修改Yolov10模型的檢測(cè)頭,使其與給定模型的檢測(cè)頭輸出一致
2. 給出更換后Yolov10模型和給定模型的性能對(duì)比,包括:準(zhǔn)確率,召回率,檢測(cè)速度
3. 完成模型在orin平臺(tái)的部署及優(yōu)化
交付需求
1. 開(kāi)發(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)證與測(cè)試代碼:
1. 模型測(cè)試代碼
2. 測(cè)試集評(píng)估結(jié)果
5. 模型導(dǎo)出代碼:導(dǎo)出成onnx格式
6. 完成Orin芯片環(huán)境部署:
1. 設(shè)置部署環(huán)境,確保依賴(lài)項(xiàng)合硬件支持
2. 部署配置與測(cè)試
7. 詳細(xì)文檔
1. 模型結(jié)果,訓(xùn)練過(guò)程,部署步驟
2. 模型測(cè)試報(bào)告