PyTorch_YOLOv4 icon indicating copy to clipboard operation
PyTorch_YOLOv4 copied to clipboard

Training on custom dataset

Open EslamHaourn opened this issue 3 years ago • 1 comments

What should I do with this piece of code when I want to train my dataset imgIds = [int(Path(x).stem) for x in dataloader.dataset.img_files] cocoGt = COCO(glob.glob('../coco/annotations/instances_val* .json')[0]) #initialize COCO ground truth api cocoDt = cocoGt.loadRes(f) # initialize COCO pred api cocoEval = COCOeval(cocoGt, cocoDt, 'bbox') cocoEval.params.imgIds = imgIds # image IDs to evaluate cocoEval.evaluate() cocoEval.accumulate() cocoEval.summarize() map, map50 = cocoEval.stats[:2] #update results ([email protected]:0.95, [email protected])

Thanks

EslamHaourn avatar Mar 08 '21 15:03 EslamHaourn

Same issue, want to train on custom dataset, what changes to make?

parthkvv avatar Sep 30 '22 05:09 parthkvv