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.

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

Layer error fix #2888

Open
wants to merge 1 commit into
base: master
Choose a base branch
from
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 4 additions & 4 deletions mrcnn/model.py
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(KL.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(KL.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(KL.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(KL.Layer):
"""Takes classified proposal boxes and their bounding box deltas and
returns the final detection boxes.

Expand Down