Skip to content

GitLab

Commit ca0477a7 authored by Gunho Park's avatar Gunho Park
Browse files

Internal change

parent 8124922d
Hide whitespace changes
Inline Side-by-side
Showing with 8 additions and 5 deletions
official/vision/beta/projects/basnet/tasks/basnet.py
View file @ ca0477a7
......@@ -237,12 +237,15 @@ class BASNetTask(base_task.Task):
loss = 0
logs = {self.loss: loss}
temp = labels['masks']
levels = sorted(outputs.keys())
logs.update({self.mae_metric.name: (labels['masks'], outputs['ref'])})
logs.update({self.maxf_metric.name: (labels['masks'], outputs['ref'])})
logs.update({self.relaxf_metric.name: (labels['masks'], outputs['ref'])})
logs.update(
{self.mae_metric.name: (labels['masks'], outputs[levels[-1]])})
logs.update(
{self.maxf_metric.name: (labels['masks'], outputs[levels[-1]])})
logs.update(
{self.relaxf_metric.name: (labels['masks'], outputs[levels[-1]])})
return logs
def inference_step(self, inputs, model):
......
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