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
4dc4f6c7
Unverified
Commit
4dc4f6c7
authored
Jan 14, 2022
by
srihari-humbarwadi
Browse files
Revert "added `kernel_size` param for `SegmentationHead`"
This reverts commit
e257b292
.
parent
09d9656f
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
0 additions
and
2 deletions
+0
-2
official/vision/beta/configs/semantic_segmentation.py
official/vision/beta/configs/semantic_segmentation.py
+0
-1
official/vision/beta/modeling/factory.py
official/vision/beta/modeling/factory.py
+0
-1
No files found.
official/vision/beta/configs/semantic_segmentation.py
View file @
4dc4f6c7
...
...
@@ -63,7 +63,6 @@ class SegmentationHead(hyperparams.Config):
num_convs
:
int
=
2
num_filters
:
int
=
256
use_depthwise_convolution
:
bool
=
False
kernel_size
:
int
=
3
prediction_kernel_size
:
int
=
1
upsample_factor
:
int
=
1
feature_fusion
:
Optional
[
...
...
official/vision/beta/modeling/factory.py
View file @
4dc4f6c7
...
...
@@ -356,7 +356,6 @@ def build_segmentation_model(
num_classes
=
model_config
.
num_classes
,
level
=
head_config
.
level
,
num_convs
=
head_config
.
num_convs
,
kernel_size
=
head_config
.
kernel_size
,
prediction_kernel_size
=
head_config
.
prediction_kernel_size
,
num_filters
=
head_config
.
num_filters
,
use_depthwise_convolution
=
head_config
.
use_depthwise_convolution
,
...
...
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