  
  
  ############################################################
-@@ -853,8 +864,9 @@ def rpn_graph(feature_map, anchors_per_location, anchor_stride):
+@@ -853,8 +864,8 @@ def rpn_graph(feature_map, anchors_per_location, anchor_stride):
                    activation='linear', name='rpn_class_raw')(shared)
  
      # Reshape to [batch, anchors, 2]

From 00cfd7d7ce323df1f71048d959681349ca4fed78 Mon Sep 17 00:00:00 2001
From: Nine Feng <nfeng@nvidia.com>
Date: Wed, 24 Jul 2019 09:22:44 +0800
Subject: [PATCH] Update the Mask_RCNN model from NHWC to NCHW

 2 files changed, 40 insertions(+), 27 deletions(-)

diff --git a/mrcnn/model.py b/mrcnn/model.py
index 67c2e4f..d297ae8 100644
--- a/mrcnn/model.py
+++ b/mrcnn/model.py
@@ -110,17 +110,17 @@ def identity_block(input_tensor, kernel_size, filters, stage, block,
