Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Update model.py #2918

Open
wants to merge 1 commit into
base: master
Choose a base branch
from
Open

Update model.py #2918

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Jump to
Jump to file
Failed to load files.
Diff view
Diff view
8 changes: 4 additions & 4 deletions mrcnn/model.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -252,7 +252,7 @@ def clip_boxes_graph(boxes, window):
return clipped


class ProposalLayer(KE.Layer):
class ProposalLayer(KE.base_layer.Layer):
"""Receives anchor scores and selects a subset to pass as proposals
to the second stage. Filtering is done based on anchor scores and
non-max suppression to remove overlaps. It also applies bounding
Expand Down Expand Up @@ -341,7 +341,7 @@ def log2_graph(x):
return tf.log(x) / tf.log(2.0)


class PyramidROIAlign(KE.Layer):
class PyramidROIAlign(KE.base_layer.Layer):
"""Implements ROI Pooling on multiple levels of the feature pyramid.

Params:
Expand Down Expand Up @@ -619,7 +619,7 @@ def detection_targets_graph(proposals, gt_class_ids, gt_boxes, gt_masks, config)
return rois, roi_gt_class_ids, deltas, masks


class DetectionTargetLayer(KE.Layer):
class DetectionTargetLayer(KE.base_layer.Layer):
"""Subsamples proposals and generates target box refinement, class_ids,
and masks for each.

Expand Down Expand Up @@ -779,7 +779,7 @@ def nms_keep_map(class_id):
return detections


class DetectionLayer(KE.Layer):
class DetectionLayer(KE.base_layer.Layer):
"""Takes classified proposal boxes and their bounding box deltas and
returns the final detection boxes.

Expand Down