Unverified Commit 7189e614 authored by srihari-humbarwadi's avatar srihari-humbarwadi
Browse files

cleanup

parent 07d7cb85
...@@ -142,21 +142,6 @@ class PanopticDeeplabTask(base_task.Task): ...@@ -142,21 +142,6 @@ class PanopticDeeplabTask(base_task.Task):
instance_center_offset_loss_fn = panoptic_deeplab_losses.CenterOffsetLoss( instance_center_offset_loss_fn = panoptic_deeplab_losses.CenterOffsetLoss(
use_groundtruth_dimension=loss_config.use_groundtruth_dimension) use_groundtruth_dimension=loss_config.use_groundtruth_dimension)
# outputs = {
# 'segmentation_outputs': segmentation_outputs,
# 'instance_centers_heatmap':
# instance_outputs['instance_centers_heatmap'],
# 'instance_centers_offset':
# instance_outputs['instance_centers_offset'],
# }
# labels = {
# 'category_mask': category_mask,
# 'instance_mask': instance_mask,
# 'instance_centers_heatmap': instance_centers_heatmap,
# 'instance_centers_offset': instance_centers_offset,
# 'things_mask': things_mask
# }
segmentation_loss = segmentation_loss_fn( segmentation_loss = segmentation_loss_fn(
model_outputs['segmentation_outputs'], model_outputs['segmentation_outputs'],
labels['category_mask']) labels['category_mask'])
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment