Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
ModelZoo
ResNet50_tensorflow
Commits
f190597d
"git@developer.sourcefind.cn:OpenDAS/mmdetection3d.git" did not exist on "0e17beabebad17a00087439c4c76748413b28908"
Commit
f190597d
authored
Jul 13, 2020
by
syiming
Browse files
removing testing code using tf1
parent
a6f9dad0
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
13 deletions
+2
-13
research/object_detection/builders/model_builder.py
research/object_detection/builders/model_builder.py
+2
-13
No files found.
research/object_detection/builders/model_builder.py
View file @
f190597d
...
@@ -72,7 +72,6 @@ if tf_version.is_tf1():
...
@@ -72,7 +72,6 @@ if tf_version.is_tf1():
from
object_detection.models
import
faster_rcnn_resnet_v1_feature_extractor
as
frcnn_resnet_v1
from
object_detection.models
import
faster_rcnn_resnet_v1_feature_extractor
as
frcnn_resnet_v1
from
object_detection.models
import
ssd_resnet_v1_fpn_feature_extractor
as
ssd_resnet_v1_fpn
from
object_detection.models
import
ssd_resnet_v1_fpn_feature_extractor
as
ssd_resnet_v1_fpn
from
object_detection.models
import
ssd_resnet_v1_ppn_feature_extractor
as
ssd_resnet_v1_ppn
from
object_detection.models
import
ssd_resnet_v1_ppn_feature_extractor
as
ssd_resnet_v1_ppn
from
object_detection.models
import
faster_rcnn_resnet_v1_fpn_keras_feature_extractor
as
frcnn_resnet_fpn_keras
from
object_detection.models.embedded_ssd_mobilenet_v1_feature_extractor
import
EmbeddedSSDMobileNetV1FeatureExtractor
from
object_detection.models.embedded_ssd_mobilenet_v1_feature_extractor
import
EmbeddedSSDMobileNetV1FeatureExtractor
from
object_detection.models.ssd_inception_v2_feature_extractor
import
SSDInceptionV2FeatureExtractor
from
object_detection.models.ssd_inception_v2_feature_extractor
import
SSDInceptionV2FeatureExtractor
from
object_detection.models.ssd_mobilenet_v2_fpn_feature_extractor
import
SSDMobileNetV2FpnFeatureExtractor
from
object_detection.models.ssd_mobilenet_v2_fpn_feature_extractor
import
SSDMobileNetV2FpnFeatureExtractor
...
@@ -230,19 +229,9 @@ if tf_version.is_tf1():
...
@@ -230,19 +229,9 @@ if tf_version.is_tf1():
frcnn_resnet_v1
.
FasterRCNNResnet152FeatureExtractor
,
frcnn_resnet_v1
.
FasterRCNNResnet152FeatureExtractor
,
}
}
FASTER_RCNN_KERAS_FEATURE_EXTRACTOR_CLASS_MAP
=
{
'faster_rcnn_resnet50_fpn_keras'
:
frcnn_resnet_fpn_keras
.
FasterRCNNResnet50FpnKerasFeatureExtractor
,
'faster_rcnn_resnet101_fpn_keras'
:
frcnn_resnet_fpn_keras
.
FasterRCNNResnet101FpnKerasFeatureExtractor
,
'faster_rcnn_resnet152_fpn_keras'
:
frcnn_resnet_fpn_keras
.
FasterRCNNResnet152FpnKerasFeatureExtractor
,
}
FEATURE_EXTRACTOR_MAPS
=
[
FEATURE_EXTRACTOR_MAPS
=
[
SSD_FEATURE_EXTRACTOR_CLASS_MAP
,
SSD_FEATURE_EXTRACTOR_CLASS_MAP
,
FASTER_RCNN_FEATURE_EXTRACTOR_CLASS_MAP
,
FASTER_RCNN_FEATURE_EXTRACTOR_CLASS_MAP
FASTER_RCNN_KERAS_FEATURE_EXTRACTOR_CLASS_MAP
]
]
...
@@ -586,7 +575,7 @@ def _build_faster_rcnn_model(frcnn_config, is_training, add_summaries):
...
@@ -586,7 +575,7 @@ def _build_faster_rcnn_model(frcnn_config, is_training, add_summaries):
_check_feature_extractor_exists
(
frcnn_config
.
feature_extractor
.
type
)
_check_feature_extractor_exists
(
frcnn_config
.
feature_extractor
.
type
)
is_keras
=
tf_version
.
is_tf2
()
is_keras
=
tf_version
.
is_tf2
()
if
is_keras
or
True
:
if
is_keras
:
feature_extractor
=
_build_faster_rcnn_keras_feature_extractor
(
feature_extractor
=
_build_faster_rcnn_keras_feature_extractor
(
frcnn_config
.
feature_extractor
,
is_training
,
frcnn_config
.
feature_extractor
,
is_training
,
inplace_batchnorm_update
=
frcnn_config
.
inplace_batchnorm_update
)
inplace_batchnorm_update
=
frcnn_config
.
inplace_batchnorm_update
)
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment