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
f00cc5da
Commit
f00cc5da
authored
Jun 14, 2020
by
syiming
Browse files
Initial commit for faster rcnn resnet v1 fpn feature extractor.
1. Setup code structure for fpn feature extractor
parent
b3ef7ae9
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
39 additions
and
0 deletions
+39
-0
research/object_detection/models/faster_rcnn_resnet_v1_fpn_keras_feature_extractor.py
...dels/faster_rcnn_resnet_v1_fpn_keras_feature_extractor.py
+39
-0
No files found.
research/object_detection/models/faster_rcnn_resnet_v1_fpn_keras_feature_extractor.py
0 → 100644
View file @
f00cc5da
"""Faster RCNN Keras-based Resnet v1 FPN Feature Extractor."""
from
object_detection.meta_architectures
import
faster_rcnn_meta_arch
from
object_detection.models
import
feature_map_generators
from
object_detection.models.keras_models
import
resnet_v1
from
object_detection.models.keras_models
import
model_utils
from
object_detection.utils
import
ops
class
FasterRCNNFPNKerasFeatureExtractor
(
faster_rcnn_meta_arch
.
FasterRCNNKerasFeatureExtractor
):
"""Faster RCNN Feature Extractor using Keras-based Resnet v1 FPN features."""
def
__init__
(
self
,
...):
# TODO: constructor
pass
def
build
(
self
,
...):
# TODO: Build the structure, should be very similar as ssd_*_fpn_keras_feature_extractor.py
# ResNet-101 (object_detection.models.keras_models)
# object_detection.models.feature_map_generators
pass
def
preprocess
(
self
,
...):
# TODO: should be the same as others
pass
def
_extract_proposal_features
(
self
,
...):
# TODO: Extracts first stage RPN features
# Fpn_feature_levels
pass
def
_extract_box_classifier_features
(
self
,
...):
# TODO: Extracts second stage box classifier features.
pass
def
restore_from_classification_checkpoint_fn
(
self
,
...):
# follow the none fpn version
pass
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