Commit c3e27db1 authored by Lamping, Christian's avatar Lamping, Christian
Browse files

Update utils.py

parent 9ac9cc73
......@@ -290,7 +290,7 @@ class Dataset(object):
for annotation in annotations:
class_id = annotation['category_id']+1
if class_id ==1: # if annotation is hen, then the score is considered, otherwise it is -1
attributes = [element for element in annotation['extra']['attributes'] if element.isdigit()]#
attributes = [element for element in annotation['extra']['attributes'] if element.isdigit()]#
if len(attributes) > 0:
score = attributes[0]
else: score = -1
......
Supports Markdown
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