"llm/git@developer.sourcefind.cn:OpenDAS/ollama.git" did not exist on "14d5093cd02373f11ed138241e3e2eedf349824b"
Commit 503cc99c authored by syiming's avatar syiming
Browse files

Fix context_rcnn_meta_arch_tf1_test.py failed test

parent 1855f881
......@@ -333,7 +333,7 @@ class ContextRCNNMetaArch(faster_rcnn_meta_arch.FasterRCNNMetaArch):
"""
box_features = self._crop_and_resize_fn(
[features_to_crop], proposal_boxes_normalized, None,
features_to_crop, proposal_boxes_normalized, None,
[self._initial_crop_size, self._initial_crop_size])
attention_features = self._context_feature_extract_fn(
......
......@@ -529,7 +529,8 @@ class ContextRCNNMetaArchTest(test_case.TestCase, parameterized.TestCase):
(rpn_box_predictor_features, rpn_box_encodings, refined_box_encodings,
proposal_boxes_normalized, proposal_boxes) = execute_fn(graph_fn, [],
graph=g)
self.assertAllEqual(rpn_box_predictor_features.shape, [2, 20, 20, 512])
self.assertAllEqual(len(rpn_box_predictor_features), 1)
self.assertAllEqual(rpn_box_predictor_features[0].shape, [2, 20, 20, 512])
self.assertAllEqual(rpn_box_encodings.shape, [2, 3600, 4])
self.assertAllEqual(refined_box_encodings.shape, [16, 42, 4])
self.assertAllEqual(proposal_boxes_normalized.shape, [2, 8, 4])
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment