tutorial code question ¶
By: N0zE on Nov. 1, 2024, 2:21 a.m.
in training ipnb What does “detection_dataset2” mean ?
I thought it is custom dataset but I don’t find any DatasetCatalog.register or register_coco_instances codes…
cfg = get_cfg()
using faster rcnn architecture
cfg.merge_from_file(
model_zoo.get_config_file("COCO-Detection/faster_rcnn_X_101_32x8d_FPN_3x.yaml")
)
cfg.DATASETS.TRAIN = ("detection_dataset2",)
cfg.DATASETS.TEST = ()
cfg.DATALOADER.NUM_WORKERS = 1
Could someone please explain this to me?