easy-Fpn源码解读(四)：dataset

本文主要是介绍easy-Fpn源码解读(四)：dataset，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

easy-Fpn源码解读(四)：dataset

base代码解析

base.py

class Base(torch.utils.data.dataset.Dataset):class Mode(Enum):TRAIN = 'train'EVAL = 'eval'OPTIONS = ['voc2007', 'coco2017', 'voc2007-cat-dog', 'coco2017-person', 'coco2017-car', 'coco2017-animal']@staticmethod    # 静态方法，不依赖于类调用，但也不能出现和该类相关的任何属性，方法def from_name(name: str) -> Type['Base']:if name == 'voc2007':from dataset.voc2007 import VOC2007return VOC2007elif name == 'coco2017':from dataset.coco2017 import COCO2017return COCO2017elif name == 'voc2007-cat-dog':from dataset.voc2007_cat_dog import VOC2007CatDogreturn VOC2007CatDogelif name == 'coco2017-person':from dataset.coco2017_person import COCO2017Personreturn COCO2017Personelif name == 'coco2017-car':from dataset.coco2017_car import COCO2017Carreturn COCO2017Carelif name == 'coco2017-animal':from dataset.coco2017_animal import COCO2017Animalreturn COCO2017Animalelse:raise ValueErrordef __init__(self, path_to_data_dir: str, mode: Mode, image_min_side: float, image_max_side: float):self._path_to_data_dir = path_to_data_dirself._mode = modeself._image_min_side = image_min_sideself._image_max_side = image_max_sidedef __len__(self) -> int:  # 属于python类定义中保留的特殊的用于重载的方法raise NotImplementedError  # raise用于实现报错功能，而NotImplementedError则是要求在子类中必须实现此方法，不然就报错def __getitem__(self, index: int) -> Tuple[str, Tensor, float, Tensor, Tensor]:raise NotImplementedErrordef evaluate(self, path_to_results_dir: str, image_ids: List[str], bboxes: List[List[float]], classes: List[int], probs: List[float]) -> Tuple[float, str]:raise NotImplementedErrordef _write_results(self, path_to_results_dir: str, image_ids: List[str], bboxes: List[List[float]], classes: List[int], probs: List[float]):raise NotImplementedError@staticmethoddef num_classes() -> int:raise NotImplementedError@staticmethod  # 关于PIL.Image：https://www.cnblogs.com/lyrichu/p/9124504.htmldef preprocess(image: PIL.Image.Image, image_min_side: float, image_max_side: float) -> Tuple[Tensor, float]:# resize according to the rules:

这篇关于easy-Fpn源码解读(四)：dataset的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---