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
0bee3f54
Unverified
Commit
0bee3f54
authored
Oct 25, 2021
by
srihari-humbarwadi
Browse files
fixed linting and docstrings
parent
9680e09d
Changes
3
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
4 additions
and
4 deletions
+4
-4
official/vision/beta/evaluation/panoptic_quality_evaluator.py
...cial/vision/beta/evaluation/panoptic_quality_evaluator.py
+1
-1
official/vision/beta/projects/panoptic_maskrcnn/configs/panoptic_maskrcnn.py
...a/projects/panoptic_maskrcnn/configs/panoptic_maskrcnn.py
+2
-2
official/vision/beta/projects/panoptic_maskrcnn/modeling/layers/paste_masks.py
...projects/panoptic_maskrcnn/modeling/layers/paste_masks.py
+1
-1
No files found.
official/vision/beta/evaluation/panoptic_quality_evaluator.py
View file @
0bee3f54
...
...
@@ -79,7 +79,7 @@ class PanopticQualityEvaluator:
instead of "stuff." Default to `None`, and it means categories are not
classified into these two categories.
rescale_predictions: `bool`, whether to scale back prediction to original
image sizes. If True,
y_true
['image_info'] is used to rescale
image sizes. If True,
groundtruths
['image_info'] is used to rescale
predictions.
"""
self
.
_pq_metric_module
=
panoptic_quality
.
PanopticQuality
(
...
...
official/vision/beta/projects/panoptic_maskrcnn/configs/panoptic_maskrcnn.py
View file @
0bee3f54
...
...
@@ -174,8 +174,8 @@ def panoptic_fpn_coco() -> cfg.ExperimentConfig:
is_thing
.
append
(
True
if
idx
<=
num_thing_categories
else
False
)
config
=
cfg
.
ExperimentConfig
(
runtime
=
cfg
.
RuntimeConfig
(
mixed_precision_dtype
=
'bfloat16'
,
enable_xla
=
True
),
runtime
=
cfg
.
RuntimeConfig
(
mixed_precision_dtype
=
'bfloat16'
,
enable_xla
=
True
),
task
=
PanopticMaskRCNNTask
(
init_checkpoint
=
'gs://cloud-tpu-checkpoints/vision-2.0/resnet50_imagenet/ckpt-28080'
,
# pylint: disable=line-too-long
init_checkpoint_modules
=
[
'backbone'
],
...
...
official/vision/beta/projects/panoptic_maskrcnn/modeling/layers/paste_masks.py
View file @
0bee3f54
...
...
@@ -131,7 +131,7 @@ class PasteMasks(tf.keras.layers.Layer):
def
__init__
(
self
,
output_size
:
List
[
int
],
grid_sampler
,
**
kwargs
):
"""
Generat
es
p
an
optic segmentation masks
.
"""
Resiz
es an
d pastes instance masks to match image size
.
Args:
output_size: A `List` of integers that represent the height and width of
...
...
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