search

yangsenius / TransPose

PyTorch Implementation for "TransPose: Keypoint localization via Transformer", ICCV 2021.
https://github.com/yangsenius/TransPose/releases/download/paper/transpose.pdf
MIT License
353 stars 56 forks source link

no modules named nms.cpu & nms.gpu #20

Closed emilyjiayaoli closed 3 years ago

emilyjiayaoli commented 3 years ago

when running python tools/test.py --cfg experiments/coco/transpose_r/TP_R_256x192_d256_h1024_enc4_mh8.yaml TEST.USE_GT_BBOX True

File ../lib/nms/nms.py", line 13, in from .cpu_nms import cpu_nms ModuleNotFoundError: No module named 'nms.cpu_nms'

Only nms.cpu.c and nms.gpu.c exists in the nms folder, not nms.cpu and nms.gpu

yangsenius commented 3 years ago

You may need to Make the lib.