switchablenorms / DeepFashion2

DeepFashion2 Dataset https://arxiv.org/pdf/1901.07973.pdf
2.29k stars 355 forks source link

Landmarks predict? #28

Open TDBTECHNO opened 5 years ago

TDBTECHNO commented 5 years ago

I have trained the model with mtcnn main.py, and check the results with matterport mask ecnyn demo My question is I am getting segmentation mask, but not the landmarks How to get landmarks on segmentation or object?

geyuying commented 5 years ago

A keypoint’s location is modeled as a one-hot mask. In this way, you can easily train landmark estimation network with your framework. Instead of predicting K masks for K categories given a detected bouding box, you need to predict K masks, one for each of K keypoint types.

TDBTECHNO commented 5 years ago

@geyuying Thanks for the suggestion

I have already trained landmark keypoints with mtcnn main.py But I would like to know how to visualize landmarks

geyuying commented 5 years ago

https://github.com/facebookresearch/Detectron/blob/master/detectron/utils/vis.py def(vis one image)

fishWangY commented 5 years ago

@geyuying Thanks for the suggestion

I have already trained landmark keypoints with mtcnn main.py But I would like to know how to visualize landmarks

can you share your code about landmark keypoints? thanks.

coordxyz commented 4 years ago

A keypoint’s location is modeled as a one-hot mask. In this way, you can easily train landmark estimation network with your framework. Instead of predicting K masks for K categories given a detected bouding box, you need to predict K masks, one for each of K keypoint types.

hi, @geyuying @TDBTECHNO , Does K equal to 294? Do I have to convert keypoints of every cloth type into 294 heatmaps as supervision? I tried to train the simple Hourglass module based network by using the converted 294 heatmaps but the network can't convergence. And I also tried OHKM. It seems that the deepfashion2_to_coco.py concatenates all 294 keypoints together for each cloth. That is, the maximum number of valid keypoints is only 39(long sleeve outwear). How to handle this unbalance between negative and positive labels?