[App4AI SDK] Jupyter-Image-Human-Pose-PyTorch

[簡介]

 

應用於人體姿勢偵測,可偵測出人的眼睛、鼻子、耳朵、脖子、肩部、手肘、手腕、髖關節、膝關節、腳踝的位置。

 

 

[操作步驟及說明]

 

解決方案大致流程為:

準備資料集 -> 產生訓練所需的檔案 -> 訓練 -> 使用各種方式進行推論

這裡所使用的資料集為 coco2017 的資料集。

 

1. 1_prepare_train_labels.ipynb

運行後會產生訓練所需的 pkl 檔。

參數:

--labels 資料集 object keypoints 格式的 json 標記檔 。 

--output-name 輸出訓練用的 pkl 檔。

補充:

若想要使用自己的資料集,需注意標記檔必須為 object keypoints 格式的 json 檔,且 train/annotations 資料夾內 object keypoints 格式的標記檔名稱必須為 “person_keypoints_train.json”,val/annotations 資料夾內 object keypoints 格式的標記檔名稱必須為 “val.json”。

 

2. 2_train.ipynb

訓練  coco2017 資料集。

參數:

--train-images-folder 訓練的影像資料夾位置

--prepared-train-labels 訓練用的 pkl 標記檔,來自 1_prepare_train_labels.ipynb 運行後產生。

--val-labels 測試影像的 object keypoints 格式 json 標記檔

--val-images-folder 測試的影像資料夾位置

--checkpoint-path pretrained 模型檔位置

 

3. 3_inference.ipynb

推論一張影像,標出人體位置及關節的位置。

參數:

--images-folder 推論影像位置。

--checkpoint-path 訓練模型位置。

 

human pose.png

 

4. 4_inference_folder.ipynb

推論資料夾內的所有影像,標出人體位置及關節的位置。

參數:

--images-folder 推論影像資料夾位置。

--checkpoint-path 訓練模型位置。

 

5. 5_inference_webcam.ipynb

推論 webcam 影像,標出人體位置及關節的位置。

參數:

--checkpoint-path 訓練模型位置。

--video 第幾個 webcam。

 

聯絡我們及購買方式


請參閱以下網址:

https://tw.leaderg.com/article/index?sn=11056

延伸閱讀

1.
App4AI 人工智慧軟體 Windows 版

2.
App4AI 人工智慧軟體 Linux 版

3.
LEADERG COMPUTER 人工智慧 GPU 電腦

4.
教育訓練課程

感謝客戶們的支持

臺灣大學、清華大學、陽明交通大學、成功大學、臺北醫學大學、臺北護理健康大學、中興大學、中山大學、暨南大學、宜蘭大學、嘉義大學、聯合大學、淡江大學、逢甲大學、長庚大學、義守大學、實踐大學、大同大學、中原大學、東吳大學、銘傳大學、臺科大、北科大、臺中科大、雲林科大、勤益科大、虎尾科大、屏東科大、高雄科大、朝陽科大、明志科大、南臺科大、亞東科大、中國科大、國防大學、海軍軍官學校、陸軍專科學校、鼓山高中、臺大醫院、榮民總醫院、長庚醫院、亞東醫院、慈濟醫院、義大醫院、中山科學研究院、國防部軍備局、法務部調查局、工研院、資策會、原子能委員會核能研究所、農業委員會特有生物研究保育中心、勞動部勞動研究所、紡織研究所、金屬工業研究發展中心、台灣儀器科技研究中心、車輛測試中心、台灣自來水公司、台積電、聯電、南亞科、華邦電子、精材科技、華信光電、友達光電、群創光電、瀚宇彩晶、台塑、台塑網、南亞塑膠、台灣化學纖維、中國石油、羅技電子、義隆電子、隆達電子、達方電子、鴻海、大聯大 - 世平興業、盟立自動化、迅得機械、南茂科技、順達科技、致伸科技、英特盛、豐興鋼鐵、中宇環保工程、竹陞科技、研揚科技、敦陽科技、竑騰科技、先豐通訊、神達電腦、神通電腦、百佳泰、科音國際、峰安車業、鴻鵠國際、藍海智能等。