ModuleNotFoundError: No module named 'crowdposetools' while running scripts for visualization

[ElviraLL]

Are we suppose to use main.py as endpoint for inference as well? I'd like to run inference on a single image (for single character) (no annotation) with provided checkpoints trained on Human-arts.

Shall I use the script you provided for inference?

export EDPOSE_COCO_PATH=/path/to/your/cocodir
export Inference_Path=/path/to/your/inference_dir
  python -m torch.distributed.launch --nproc_per_node=1  main.py \
 --output_dir "logs/coco_r50" \
 -c config/edpose.cfg.py \
 --options batch_size=1 epochs=60 lr_drop=55 num_body_points=17 backbone='resnet50' \
 --dataset_file="coco" \
 --pretrain_model_path "./models/edpose_r50_coco.pth" \
 --eval

What's the purpose of ESPOSE_COCO_PATH variable, do we still need it if we only want to run inference? is Inference_Path suppose to contain the testing images that I'd like to try for inference? Why do we need to provide espochs and lr_drop for inference? Will the output visualization results with keypoints be saved in output_dir?

When I try the code, I didn't update the ESPOSE_COCO_PATH, I use a folder with input image for Inference_Path I also changed output_dir.

export Inference_Path=/path/to/your/input_image_folder
  python -m torch.distributed.launch --nproc_per_node=1  main.py \
 --output_dir "logs/coco_r50" \
 -c config/edpose.cfg.py \
 --options batch_size=1 epochs=60 lr_drop=55 num_body_points=17 backbone='resnet50' \
 --dataset_file="coco" \
 --pretrain_model_path "./models/edpose_r50_coco.pth" \
 --eval

However, I got the following error

Traceback (most recent call last):
  File "main.py", line 15, in <module>
    import datasets
  File "/home/jing/Documents/projects/ed-pose/datasets/__init__.py", line 4, in <module>
    from .crowdpose import build as build_crowdpose
  File "/home/jing/Documents/projects/ed-pose/datasets/crowdpose.py", line 12, in <module>
    from crowdposetools.coco import COCO
ModuleNotFoundError: No module named 'crowdposetools'

Thank you!!!!!