1條Prompt搞定AI摔倒檢測
哈嘍,大家好。
之前給大家分享過用骨骼點動作序列+STGCN(時空圖卷積神經(jīng)網(wǎng)絡)來識別摔倒。
圖片
還分享過用ChatGPT一步一步地用CNN + RNN訓練視頻動作分類器來識別摔倒。
這次我們繼續(xù)分享摔倒檢測項目,今天分享的是用一條prompt直接搞定整個項目。
圖片
數(shù)據(jù)集、代碼和prompt已經(jīng)準備好,需要的朋友留言區(qū)留言即可。
當然,這一條prompt不是一簇而就得,也不是憑空想象的,而是經(jīng)過跟gpt4進行許多次交流,不斷總結,不斷生成代碼,根據(jù)BUG不斷修改,一點點調(diào)試出來的。
下面我分享下這個思路。
先給gpt4一個簡單的設定,向給他交代任務
圖片
這時候gpt4會告訴你完成這個項目所需要的步驟。
接下里,你就需要把每一步中的內(nèi)容總結、提煉,一點點加入到你最終要形成的prompt中。
這里面每一步還不具體,gpt4只給你了一些選項,你需要繼續(xù)跟gpt4交流,確定具體的方案,將其加入到你的prompt中
圖片
經(jīng)過幾輪大致的交流,你的prompt已經(jīng)有了雛形,這時候你可以把它直接發(fā)給gpt4讓他生成完整的代碼。
前N次不要抱希望,代碼會有各種錯誤,尤其是每一步之間代碼的銜接不是很好。
這時候你需要找到gpt4生成代碼中的問題,然后修改prompt,對gpt4生成的代碼進行約束、限制,保證他生成的代碼是按照你要求的邏輯來的。
從描述數(shù)據(jù)集的格式、到數(shù)據(jù)預處理、到提取視頻特征、再到訓練GRU模型,整個過程都可以在一條prompt里搞定。