{"action":"created","issue":{"url":"https://api.github.com/repos/Superlee506/Mask_RCNN_Humanpose/issues/1","repository_url":"https://api.github.com/repos/Superlee506/Mask_RCNN_Humanpose","labels_url":"https://api.github.com/repos/Superlee506/Mask_RCNN_Humanpose/issues/1/labels{/name}","comments_url":"https://api.github.com/repos/Superlee506/Mask_RCNN_Humanpose/issues/1/comments","events_url":"https://api.github.com/repos/Superlee506/Mask_RCNN_Humanpose/issues/1/events","html_url":"https://github.com/Superlee506/Mask_RCNN_Humanpose/issues/1","id":309208672,"number":1,"title":"How to change the code to train my own dataset?","user":{"login":"zhoulukuan","id":16110591,"avatar_url":"https://avatars2.githubusercontent.com/u/16110591?v=4","gravatar_id":"","url":"https://api.github.com/users/zhoulukuan","html_url":"https://github.com/zhoulukuan","followers_url":"https://api.github.com/users/zhoulukuan/followers","following_url":"https://api.github.com/users/zhoulukuan/following{/other_user}","gists_url":"https://api.github.com/users/zhoulukuan/gists{/gist_id}","starred_url":"https://api.github.com/users/zhoulukuan/starred{/owner}{/repo}","subscriptions_url":"https://api.github.com/users/zhoulukuan/subscriptions","organizations_url":"https://api.github.com/users/zhoulukuan/orgs","repos_url":"https://api.github.com/users/zhoulukuan/repos","events_url":"https://api.github.com/users/zhoulukuan/events{/privacy}","received_events_url":"https://api.github.com/users/zhoulukuan/received_events","type":"User","site_admin":false},"labels":[],"state":"open","locked":false,"assignee":null,"assignees":[],"milestone":null,"comments":0,"created_at":"2018-03-28T03:30:19Z","updated_at":"2018-03-28T03:47:38Z","closed_at":null,"author_association":"NONE","body":"Thanks for your great work. In fact, I want to use mask rcnn to predict keypoints in clothes which have more classes and keypoints. I have two questions:\r\n1. Due to the different formats of train/test data, I must change it to the format of COCO annotations, right?\r\n2. More classes and keypoints lead to different networks and the ways of processing. Do I need to change the code of network/loss or other places? Can you give me some help?\r\nThank you in advance."},"comment":{"url":"https://api.github.com/repos/Superlee506/Mask_RCNN_Humanpose/issues/comments/376752145","html_url":"https://github.com/Superlee506/Mask_RCNN_Humanpose/issues/1#issuecomment-376752145","issue_url":"https://api.github.com/repos/Superlee506/Mask_RCNN_Humanpose/issues/1","id":376752145,"user":{"login":"Superlee506","id":13114675,"avatar_url":"https://avatars3.githubusercontent.com/u/13114675?v=4","gravatar_id":"","url":"https://api.github.com/users/Superlee506","html_url":"https://github.com/Superlee506","followers_url":"https://api.github.com/users/Superlee506/followers","following_url":"https://api.github.com/users/Superlee506/following{/other_user}","gists_url":"https://api.github.com/users/Superlee506/gists{/gist_id}","starred_url":"https://api.github.com/users/Superlee506/starred{/owner}{/repo}","subscriptions_url":"https://api.github.com/users/Superlee506/subscriptions","organizations_url":"https://api.github.com/users/Superlee506/orgs","repos_url":"https://api.github.com/users/Superlee506/repos","events_url":"https://api.github.com/users/Superlee506/events{/privacy}","received_events_url":"https://api.github.com/users/Superlee506/received_events","type":"User","site_admin":false},"created_at":"2018-03-28T03:47:38Z","updated_at":"2018-03-28T03:47:38Z","author_association":"OWNER","body":"Mask RCNN for human pose estimation just has 2 classes(1 for background 1 for person). So as to your question. You just need to change  the following functions:\r\n1. In fpn_classifier_graph, you just need to change the num_classes\r\n2. In build_fpn_mask_graph, you also just need to change the num_classes\r\n3. In build_fpn_keypoint_graph, because I use it to predict the keypoint of human, you have to change the output as [batch_size, num_roi, num_classes, num_keypoints].\r\n4. In keypoint_mrcnn_mask_loss_graph, you need to add some code to get the class specific mask of each ROI as mrcnn_mask_loss_graph does."}}
{"actor":{"display_login":"Superlee506"}}