bubbliiiing / mask-rcnn-keras 训练自己的数据集

code:https://github.com/bubbliiiing/mask-rcnn-keras
blog:

步骤：
１．修改train.py和dataset.py类别
２．labelme标注数据集放到train_dataset目录

问题：
１．module ‘tensorflow.math’ has no attribute ‘is_nan’
方法：
不同版本的tensorflow结构有差别。
原代码是tf1.13的，我是tf1.11.0，在mrcnn_training.py中要改连个函数：

    # loss = K.switch(tf.math.is_nan(loss), tf.constant([0.0]), loss)
    loss = K.switch(tf.is_nan(loss), tf.constant([0.0]), loss)

另外比如同一个函数，在1.9中：
import tensorflow as tf
x = tf.bincount(x)
而在1.13中：
import tensorflow as tf
x = tf.math.bincount(x)