|
@@ -2,14 +2,12 @@ import abc
|
|
|
import logging
|
|
|
import typing as T
|
|
|
|
|
|
-from collections import namedtuple
|
|
|
from cvdatasets import AnnotationType
|
|
|
+from cvdatasets import AnnotationArgs
|
|
|
from cvdatasets.dataset.image import Size
|
|
|
|
|
|
from cvfinetune.finetuner.mixins.base import BaseMixin
|
|
|
|
|
|
-Opts = namedtuple("Opts", "data dataset parts feature_model")
|
|
|
-
|
|
|
class _DatasetMixin(BaseMixin):
|
|
|
"""
|
|
|
This mixin is responsible for annotation loading and for
|
|
@@ -55,7 +53,7 @@ class _DatasetMixin(BaseMixin):
|
|
|
Reads annotations and creates annotation instance,
|
|
|
which holds important infos about the dataset
|
|
|
"""
|
|
|
- opts = Opts(
|
|
|
+ opts = AnnotationArgs(
|
|
|
self.info_file,
|
|
|
self.dataset_name,
|
|
|
self.part_type,
|