Adding more classes

[trocolit2]

Dear @bborja

Thanks for sharing your work and code. We have tested it with different images and situations and seems robust, giving us the expected results.

I would like to extend the number of classes, to include the boat and buoy.
Is it possible to include these 2 new classes? Which approach do you suggest to follow?

Regards