search

NVlabs / CenterPose

Single-Stage Keypoint-based Category-level Object Pose Estimation from an RGB Image (ICRA 2022)
Other
285 stars 34 forks source link

I find some suspicious code in decode process #21

Open BboyHanat opened 1 year ago

BboyHanat commented 1 year ago

In decode.py line 188:
"mask_2 = (mask_2 == 7).float().expand(batch, num_joints, K, 2)"

but when I debug this code, there are all "False" value, since mask_2 is a bool tensor(also, my pytorch vision is 2.0.0, maybe it's correct in lower vision)

AnukritiSinghh commented 1 year ago

were you able to correct this?