[LEADERG AI ZOO] 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。

 


延伸閱讀

1.
LEADERG AOI VISION 人工智能影像分析軟體 (No Code, 不用寫程式, 視窗操作介面, 易學易用, 無經驗可, Deep Learning, 深度學習, Machine Learning, 機器學習)

2.
LEADERG AI ZOO 人工智能演算法軟體 (Low Code, 整理及優化190種範例程式, 節省90%的開發時間, 10倍速快速開發 AI 程式, Jupyter Lab 操作介面, AI Model Zoo, 深度學習, 機器學習, Big Data, 大數據)

3.
LEADERG ACE 人工智能電腦 (工作站、伺服器、筆電,GPU, computer, workstation, notebook, server, NVIDIA, RTX-3090-24G, RTX-A6000-48G, A100-40G, deep learning, 深度學習, machine learning, 機器學習)

4.
LEADERG EDU 人工智能教育訓練課程 (影像分析, 影像分類, 物件偵測, 影像切割, 資料分析, deep learning, 深度學習, machine learning, 機器學習, big data, 大數據)

感謝客戶們的支持

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