This commit is contained in:
Joseph 2020-10-27 14:48:11 +05:30
parent 87fd0069e2
commit 39944f9da5

View File

@ -90,13 +90,11 @@ def load_voc_coco_instances(dirname: str, split: str, class_names: Union[List[st
if cls_name not in known_class_list:
continue
if len(fileid) >= 10:
if 'unk' in split:
cls = "unknown"
else:
cls = cls_name
else:
cls = cls_name
# if 'unk' in split:
# cls = "unknown"
# else:
# cls = cls_name
cls = cls_name
# We include "difficult" samples in training.
# Based on limited experiments, they don't hurt accuracy.
# difficult = int(obj.find("difficult").text)