Files
fabric-point-cloud/3dscan/__pycache__/skeleton.cpython-311.pyc
T

33 lines
4.0 KiB
Plaintext
Raw Normal View History

2024-08-18 11:28:27 -06:00
§
WnÁf ãó¶ddlZddlZddlZddlZddlmZdZej j
Z ej j Z
e  ddd¬¦«Z
dZdZd „Zd
Zd
d ZdS)éN)Ú
ConvexHullTFgà?)Ústatic_image_modeÚmin_detection_confidenceÚmin_tracking_confidencecóJt tj¦«S©N)ÚkÚ get_frameÚktbÚCOLOR©óúE/home/astatin3/Documents/GitHub/fabric-point-cloud/3dscan/skeleton.pyÚ get_colorróÝ ;Š;•s”yÑ !rcóJt tj¦«Sr)r r
r ÚDEPTHr
rrÚ
get_depth_maprrrcó†tj|¦«\}}}}tj|d||z
z | dz||z
z ¬¦«}|S)Ngào@)ÚalphaÚbeta)Úcv2Ú minMaxLocÚconvertScaleAbs)Ú depth_mapÚmin_valÚmax_valÚ
normalizeds rÚnormalize_depth_mapr sVÝ œ]¨9Ñ€GˆWÝÔ$ Y°e¸WÀW¹_Ñ6MÐU\ÐT\Ð_dÑTdÐfmÐnuÑfuÑTvÐw€JØ Ðrcóòt |¦«}|jr*td¦«t ||j|j¦«St
j|jddt j¬¦«S)Nz
Found person!é©Údtype) ÚposeÚprocessÚpose_landmarksÚprintÚcreate_person_maskÚshapeÚnpÚzerosÚuint8)rÚ rgb_imageÚresultss rÚ calc_maskr0sgõlŠl˜9Ñ%€GàÔð
ˆoÑÔÐÝ! )¨WÔ-CÀYÄ_ÑŒx˜ œ¨¨¨Ô+µ2´<réc óîtj|ddtj¬¦«}tjD]¼}|j|d}|j|d}t
|j|dz¦«t
|j|dz¦«} }t
|j|dz¦«t
|j|dz¦«} }
tj
||| f|
| fd|dz¬¦«Œ½tj ||ftj¦«} tj || d¬¦«}g}
|jD]…}t
|j|dz¦«t
|j|dz¦«}}d|cxkr |dkr6nŒRd|cxkr |dkr nŒh|
 
|||f¦«Œ†tj|
d¦«}tj|
d¦«}||k||kz}||z}|S)Nr"r#ré)Ú thickness)Ú
iterationséd)r+r,r-Úmp_poseÚPOSE_CONNECTIONSÚlandmarkÚintÚyrÚlineÚonesÚdilateÚappendÚ
percentile)rr'Ú image_shapeÚdistance_thresholdÚmaskÚ