Train Mask R-CNN Net for Object Detection in 60 Lines of Code

Now we can start writing the code. First, let's import packages and define the main training parameters: All images during the training processes will be resized to this size. Depending on your hardware, it might be necessary to use a smaller batchSize or image size to avoid out-of-memory problems. Note that since we train with only a single image size, the net once trained is likely to be limited to work with only images of this size. In most cases what you want to do is change the size of each training batch.