search

Naman-ntc / Pytorch-Human-Pose-Estimation

Implementation of various human pose estimation models in pytorch on multiple datasets (MPII & COCO) along with pretrained models
MIT License
474 stars 75 forks source link

No module named 'datasets.COCO.nms.cpu_nms' #17

Closed ShuangjunLiu closed 3 years ago

ShuangjunLiu commented 4 years ago

I followed the instruction and tried to run a test.
However, something seems to be wrong with coco api.
I have installed coco api before which was working. I used it in my other work before.

Traceback (most recent call last): File "main.py", line 2, in import builder File "/scratch/liu.shu/codesPool/Pytorch-Human-Pose-Estimation/builder.py", line 5, in import dataloaders as dataloaders File "/scratch/liu.shu/codesPool/Pytorch-Human-Pose-Estimation/dataloaders.py", line 1, in import datasets File "/scratch/liu.shu/codesPool/Pytorch-Human-Pose-Estimation/datasets/init.py", line 2, in from datasets.coco import * File "/scratch/liu.shu/codesPool/Pytorch-Human-Pose-Estimation/datasets/coco.py", line 2, in from datasets.COCO.coco import COCODataset File "/scratch/liu.shu/codesPool/Pytorch-Human-Pose-Estimation/datasets/COCO/init.py", line 11, in from .coco import COCODataset as coco File "/scratch/liu.shu/codesPool/Pytorch-Human-Pose-Estimation/datasets/COCO/coco.py", line 24, in from .nms.nms import oks_nms File "/scratch/liu.shu/codesPool/Pytorch-Human-Pose-Estimation/datasets/COCO/nms/nms.py", line 13, in from .cpu_nms import cpu_nms ModuleNotFoundError: No module named 'datasets.COCO.nms.cpu_nms'

amin-asdzdh commented 4 years ago

I encountered the same problem, have you figured out how issue? Cheers

ShuangjunLiu commented 4 years ago

I encountered the same problem, have you figured out how issue? Cheers

I thought it missed the necessary Makefile for nms.
I assumed this part was originated from this repo. You can use their nms instead.

Naman-ntc commented 4 years ago

Yes I believe @ShuangjunLiu is right. You need to build nms for COCO dataset as per the instructions here