From 27e4a78d4ed363c7dad42fd2c140746c7425cfc8 Mon Sep 17 00:00:00 2001 From: CharlesShang Date: Tue, 7 Mar 2017 13:36:49 +0800 Subject: [PATCH] fixing bugs when trans to TF1.0 --- lib/roi_pooling_layer/roi_pooling_op_grad.py | 1 + 1 file changed, 1 insertion(+) diff --git a/lib/roi_pooling_layer/roi_pooling_op_grad.py b/lib/roi_pooling_layer/roi_pooling_op_grad.py index 8f03f7e..262efcf 100644 --- a/lib/roi_pooling_layer/roi_pooling_op_grad.py +++ b/lib/roi_pooling_layer/roi_pooling_op_grad.py @@ -1,5 +1,6 @@ import tensorflow as tf from tensorflow.python.framework import ops +import roi_pooling_op @ops.RegisterGradient("RoiPool") def _roi_pool_grad(op, grad, _):