Skip to content

GitLab

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

Internal change

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