yangsenius / TransPose

PyTorch Implementation for "TransPose: Keypoint localization via Transformer", ICCV 2021.

Home Page:https://github.com/yangsenius/TransPose/releases/download/paper/transpose.pdf

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

no modules named nms.cpu & nms.gpu

emilyjiayaoli opened this issue · comments

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

You may need to Make the lib.