keypoint_rcnn_training_pytorch icon indicating copy to clipboard operation
keypoint_rcnn_training_pytorch copied to clipboard

Training on multi classes

Open sainemustas321 opened this issue 2 years ago • 3 comments

Hi, first of all. Thank you for sharing your excellent work. I tried to modify a code to train a multi classes detection. In this line. Does it allow to change to my own multi classes??? thank you. bboxes_labels_original = ['Glue tube' for _ in bboxes_original]

sainemustas321 avatar Mar 11 '23 16:03 sainemustas321

Hi, first of all. Thank you for sharing your excellent work. I tried to modify a code to train a multi classes detection. In this line. Does it allow to change to my own multi classes??? thank you. bboxes_labels_original = ['Glue tube' for _ in bboxes_original]

Hi, did you find the way to solve this problem? I have an idea to solve two problems (object classification and keypoint detection) by two different algorithms,. But of course one would be better ;)

WojtiF avatar Apr 01 '23 16:04 WojtiF

if you use two algorithms, the time would be influenced.

MrZhang-YUBO avatar Apr 20 '23 03:04 MrZhang-YUBO

Of course, but it's my task to use neural network for classification and localization with orientation (I want to use keypoint detection for this). Even if it cause longer time.

The best option would be one algorithm but I can't find solution for this.

WojtiF avatar Apr 20 '23 13:04 WojtiF