Skip to content

GitLab

"driver/vscode:/vscode.git/clone" did not exist on "e2b4c5b469db934af4b8b81d860df56be69764ce"
Commit 1538da90 authored by syiming's avatar syiming
Browse files

modify feature extractor name

parent aef04705
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
research/object_detection/models/faster_rcnn_resnet_v1_fpn_keras_feature_extractor_tf2_test.py
View file @ 1538da90
...@@ -26,7 +26,7 @@ from object_detection.protos import hyperparams_pb2 ...@@ -26,7 +26,7 @@ from object_detection.protos import hyperparams_pb2
@unittest.skipIf(tf_version.is_tf1(), 'Skipping TF2.X only test.') @unittest.skipIf(tf_version.is_tf1(), 'Skipping TF2.X only test.')
class FasterRCNNResnetV1FPNKerasFeatureExtractorTest(tf.test.TestCase): class FasterRCNNResnetV1FpnKerasFeatureExtractorTest(tf.test.TestCase):
def _build_conv_hyperparams(self): def _build_conv_hyperparams(self):
conv_hyperparams = hyperparams_pb2.Hyperparams() conv_hyperparams = hyperparams_pb2.Hyperparams()
...@@ -44,7 +44,7 @@ class FasterRCNNResnetV1FPNKerasFeatureExtractorTest(tf.test.TestCase): ...@@ -44,7 +44,7 @@ class FasterRCNNResnetV1FPNKerasFeatureExtractorTest(tf.test.TestCase):
return hyperparams_builder.KerasLayerHyperparams(conv_hyperparams) return hyperparams_builder.KerasLayerHyperparams(conv_hyperparams)
def _build_feature_extractor(self): def _build_feature_extractor(self):
return frcnn_res_fpn.FasterRCNNResnet50FPNKerasFeatureExtractor( return frcnn_res_fpn.FasterRCNNResnet50FpnKerasFeatureExtractor(
is_training=False, is_training=False,
conv_hyperparams=self._build_conv_hyperparams(), conv_hyperparams=self._build_conv_hyperparams(),
first_stage_features_stride=16, first_stage_features_stride=16,
......
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