GengDavid / pytorch-cpn

A PyTorch re-implementation of CPN (Cascaded Pyramid Network for Multi-Person Pose Estimation)
GNU General Public License v3.0
484 stars 98 forks source link

About using myself val_det #2

Closed xiashh closed 6 years ago

xiashh commented 6 years ago

If I want to use myself detection results, should I transform the annotations to some particular format? I have read your code, and it seems that it's not mentioned.

GengDavid commented 6 years ago

Yes, if you want to use your own detection results, you'll need to transform your results into a json file. It should contain only one list of data. Each list item represents one detection result, and the format of it is like this:

{
    "unit": {
        "GT_bbox": [x1, y1, x2, y2]
    },
    "imgInfo": {
        "imgID": int,
        "img_paths": str,
    },
    "score": float
}

For example, for test image '000000581781.jpg' in test set, the imgID is 581781 and img_paths is '000000581781.jpg'. And score is the detection confidence value.
The format may be a little bit strange because I only use ground truth bbox to test at the very beginning.

xiashh commented 6 years ago

Thank you very much! I'm a novice for deep learning. So may I ask you about implementation pipeline? I don't know where to begin.

GengDavid commented 6 years ago

This repo is implemented follow the implementation of tf-cpn.
And there is actually no fixed pipeline. Generally, you can build your network up first, and then you can train it with your own processed data. You can refer to others' hyper-parameter first to train your own networks.
Have fun with it.

xiashh commented 6 years ago

Thank you!