|
@@ -1,4 +1,3 @@
|
|
|
-import logging
|
|
|
import typing
|
|
|
|
|
|
from cvdatasets.annotation.types.file_list import FileListAnnotations
|
|
@@ -6,7 +5,6 @@ from cvdatasets.annotation.types.folder_annotations import FolderAnnotations
|
|
|
from cvdatasets.annotation.types.json_annotations import JSONAnnotations
|
|
|
|
|
|
from cvargparse.utils import BaseChoiceType
|
|
|
-from cvargparse.utils.enumerations import MetaBaseType
|
|
|
from cvdatasets.utils import read_info_file
|
|
|
|
|
|
|
|
@@ -35,7 +33,7 @@ class AnnotationType(BaseChoiceType):
|
|
|
|
|
|
else:
|
|
|
assert args.dataset in info_file.DATASETS, \
|
|
|
- f"No information was found about the dataset \"{opts.dataset}\" in the info file \"{opts.data}\""
|
|
|
+ f"No information was found about the dataset \"{args.dataset}\" in the info file \"{args.data}\""
|
|
|
annot = cls[ds_info.annotation_type.lower()].value
|
|
|
|
|
|
return annot.new(args, ds_info=ds_info, **kwargs)
|