Semantic Segmentation Suite in TensorFlow. Implement, train, and test new Semantic Segmentation models easily!
2.51k
stars
879
forks
source link
Pass argument pretrained_dir in build_refinenet to frontend_builder #109
Closed
msseibel closed 6 years ago
Please specify the following information when submitting an issue:
Describe the problem
Something between bug and feature: See the linked line below: https://github.com/GeorgeSeif/Semantic-Segmentation-Suite/blob/4b7eb0c1a1622367a4378cfd42d3b5b621bcb3e8/models/RefineNet.py#L158 The argument
pretrained_dir
from the methodbuild_refinenet
is not passed towards the methodfrontend_builder.build_frontend
and therefore currently useless.The argument
pretrained_dir
is also not passed in https://github.com/GeorgeSeif/Semantic-Segmentation-Suite/blob/4b7eb0c1a1622367a4378cfd42d3b5b621bcb3e8/models/DeepLabV3_plus.py#L75Are there any reasons for not passing it?
My suggestion:
logits, end_points, frontend_scope, init_fn = frontend_builder.build_frontend(inputs, frontend, is_training=is_training,pretrained_dir=pretrained_dir)