Could you give some suggestions about how to train on my own dataset

[scut-salmon]

How can I obtain my own pickle files in VGG/models base on my dataset ?

[gooa1121]

Hi, I pushed some files to train new models on different dataset. You can take a look at the guide and try the files mentioned in it. Let me know if you have any error or problem when trying it.

[lc-git]

@gooa1121 Thank you for your work. When retraining the Gaussian and regression model, what labels do I need to use? The body parts mask? like background subtraction retraining?

[gooa1121]

Following the section 3.2.2 in the main paper, the label is the normalized displacement between the coordinate of the pixel and the center of the bounding box it belongs to. For the pixels not belonging to any bounding box, you can give it any number <= -1

[lc-git]

As you mentioned, label is the displacement between the pixel and the center so I think label has 2 channels. Why the shape of parts labels is (H, W, 5)? What does "5" mean? Thank you.

[gooa1121]

The first channel indicates foreground or background, 2nd and 3rd are the displacement to the center, 4th and 5th are the displacement to the right, bottom corner of the bounding box.