search

mertkayhan / SSL-2D-Pose

Code for the paper "Deep Attention Based Semi-Supervised 2D-Pose Estimation for Surgical Instruments"
12 stars 3 forks source link

extract_frames.py #1

Open BlueShit-1120 opened 3 years ago

BlueShit-1120 commented 3 years ago

Dear Mert, I met an error with extract_frames.py with EndoVis dataset. how to generate the .npy file of labels in the label_dir. Could you give me some advice? Thanks.

mertkayhan commented 3 years ago

Hello Xiaowen,

you can check https://github.com/mertkayhan/SSL-2D-Pose/blob/master/utils/generate_heatmap.py to see how heatmaps are computed. As for the ground truth annotation we used:

https://github.com/surgical-vision/EndoVisPoseAnnotation https://sites.google.com/site/sznitr/code-and-datasets

for EndoVis and RMIT datasets respectively.

Best regards, Mert

BlueShit-1120 commented 3 years ago

OK. Thanks for your reply.