[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。

 

聯絡我們及購買方式


請參閱以下網址:

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


延伸閱讀

1.
LEADERG App4AI 人工智能軟體 (App for AI, 不用寫程式, 視窗操作介面, 易學易用, 無經驗可, AI Model Zoo, No Code, Deep Learning, 深度學習, Machine Learning, 機器學習, 人工智慧)

2.
LEADERG AI Zoo 人工智能演算法 (整理及優化200多種範例程式碼, 節省90%的開發時間, 10倍速快速開發 AI 程式, Jupyter Lab 操作介面, Code for AI, Code4AI, Less Coding, AI Model Zoo, AIZOO, 深度學習, Big Data, 大數據)

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

4.
LEADERG AI Education 人工智能教育 (AI 教育, 顧問服務, 影像分析, 影像分類, 物件偵測, 影像切割, 資料分析, deep learning, 深度學習, machine learning, 機器學習, big data, 大數據)

感謝客戶們的支持

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