Train Mask R-CNN Net for multiclass object detection with PyTorch

141 Views Asked by At

I want to use the code given at this tutorial here and extend it to mulitclass-object detection.

Unfortunately, the tutorial does not explain how to do that.

What I want for a result is this picture: here.

So, I guess, I somehow have to include the vessel class labels from the JSON-Files of the LabPics_V2 dataset into the training process, but how?

As I am quite new to PyTorch and python programming, I need some help for this task. If you have any ideas or suggestions, please let me know.

Thanks a lot!

0

There are 0 best solutions below