From 236234a0272f96c31b43ef3d079e26bf2385c5b6 Mon Sep 17 00:00:00 2001 From: neophack Date: Tue, 24 May 2022 19:10:14 +0800 Subject: [PATCH] Update utils.py --- omnidet/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/omnidet/utils.py b/omnidet/utils.py index 155c7fc..65181ae 100644 --- a/omnidet/utils.py +++ b/omnidet/utils.py @@ -240,7 +240,7 @@ def __init__(self, num_classes, dataset, ignore_index=None): self.dataset = dataset self.classes = dict(woodscape_raw=["void", "road", "lanemarks", "curb", "person", - "rider", "vehicles", "bicycle", "motorcycle""traffic_sign"], + "rider", "vehicles", "bicycle", "motorcycle", "traffic_sign"], motion=['static', 'motion'], ) self.reset()