1.算力市場(chǎng)租用主機(jī)
AutoDL
選擇基礎(chǔ)鏡像
創(chuàng)建之后
2.上傳文件
點(diǎn)擊開機(jī),也可在更多里面選擇無(wú)卡模式開機(jī)(此模式不能訓(xùn)練,但是可以上傳文件且更便宜)。開機(jī)之后,上傳代碼可通過(guò)xshell工具或者可以通過(guò)快捷工具JupyterLab。我兩種方法都來(lái)演示一遍。yolov5代碼
1.通過(guò)xshell上傳代碼
復(fù)制登錄指令
原指令
ssh -p 26812 root@region-8.seetacloud.com
修改為
ssh root@region-8.seetacloud.com 26812
回車后會(huì)要求輸入密碼,將密碼復(fù)制過(guò)來(lái)就可進(jìn)入
再通過(guò)快捷鍵ctrl+Alt+F通過(guò)xftp上傳代碼就可以了
2.通過(guò)JupyterLab上傳代碼
點(diǎn)擊JupyterLab后會(huì)進(jìn)入啟動(dòng)頁(yè),點(diǎn)擊箭頭所指的上傳代碼,這里我代碼已經(jīng)上傳上去了。
注意:這里上傳文件只能上傳單個(gè)文件,所有你要將代碼打個(gè)壓縮包后上傳,上傳壓縮包后需要解壓,點(diǎn)擊終端后輸入以下解壓命令
unzip xxx.zip
解壓后就會(huì)出現(xiàn)圖中的yolov5-7.0文件夾
同時(shí)也將數(shù)據(jù)集也上傳上去,如圖中的dataset,數(shù)據(jù)集我是放在根目錄的,和yolov5同級(jí)。
然后進(jìn)入yolov5這個(gè)層級(jí),輸入以下命令導(dǎo)入requirement.txt的依賴。
pip install -r requirements.txt
3.修改代碼
導(dǎo)入依賴完成后,就開始修改代碼。以下是我需要用到的yaml文件修改的代碼,我的類別數(shù)是5,以及修改后圖片的路徑。(最好在本地修改完成后再上傳上去)
train.py
模型訓(xùn)練需要使用預(yù)訓(xùn)練權(quán)重,大家可以去官網(wǎng)下載,這里我使用的是yolov5s.pt這個(gè)權(quán)重文件
4.開始訓(xùn)練
輸入python train.py開始訓(xùn)練。
注意:如果你是無(wú)卡模式開機(jī)訓(xùn)練的話,進(jìn)程會(huì)被Kill掉,需要重新開機(jī)。
文章來(lái)源:http://www.zghlxwxcb.cn/news/detail-450363.html
5.結(jié)語(yǔ)
這個(gè)平臺(tái)還是挺便宜的,如果你是學(xué)生的話還可以認(rèn)證學(xué)生,有優(yōu)惠。
另外建議大家使用xshell結(jié)合xftp,方便操作。訓(xùn)練完成后保存的模型.pt文件不能直接下載下來(lái),最好使用xftp。文章來(lái)源地址http://www.zghlxwxcb.cn/news/detail-450363.html
到了這里,關(guān)于通過(guò)AutoDL使用yolov5.7訓(xùn)練自己的數(shù)據(jù)集的文章就介紹完了。如果您還想了解更多內(nèi)容,請(qǐng)?jiān)谟疑辖撬阉鱐OY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章,希望大家以后多多支持TOY模板網(wǎng)!