I
I
Ilya Pavlyukov2017-07-31 23:31:23
Python
Ilya Pavlyukov, 2017-07-31 23:31:23

What neural network architecture to choose to recognize objects by mask?

I have a large dataset, such images (5000x5000px) and masks for them:
77c48096c0f7466fb33e6c4c3f2b5fbd.PNG6f53ee5b1105411d97697159b5ec7d5a.PNG
Is there any architecture or ready-made implementation of a neural network that could generate such a mask yourself in the future. The catch is that the standard implementation of R-CNN only recognizes rectangular regions defined through two points (I'll be wrong).
The closest thing for me is Python->Keras, but other tools will do.
UPD: The answer below solves the problem perfectly. As an alternative, I look towards Pix2Pix .

Answer the question

In order to leave comments, you need to log in

1 answer(s)
A
Arseny Kravchenko, 2017-08-01
@ilya310300

U-net , there are tons of ready-made implementations.
Train on random crops, predict by window.

Didn't find what you were looking for?

Ask your question

Ask a Question

731 491 924 answers to any question