摘要: train.py里make_dataloader读取数据集。 train_loader = make_dataloader(cfg, is_train=True, distributed=args.distributed) dataset build.py中的39行 dataset = eval(c 阅读全文
posted @ 2023-02-01 23:16 祥瑞哈哈哈 阅读(101) 评论(0) 推荐(0)
摘要: DEKR/lib/config/__init__.py中有from .default import _C as cfg。 DEKR/lib/config/default.py中的。_C更像参数 from yacs.config import CfgNode as CN _C = CN() _C.OU 阅读全文
posted @ 2023-02-01 18:40 祥瑞哈哈哈 阅读(20) 评论(0) 推荐(0)
摘要: class CocoTrainDataset(Dataset): def __init__(self, labels, images_folder, stride, sigma, paf_thickness, transform=None): super().__init__() self._ima 阅读全文
posted @ 2023-02-01 11:08 祥瑞哈哈哈 阅读(17) 评论(0) 推荐(0)