Pruning based on self built datasets

[rememberBr]

Hello, may I ask if it is possible to add a pruning method on a self built dataset? Currently, the dataset can only select ['VOC ',' COCO ']

[HankYe]

Hi, thanks for ur attention to our work. We have tested the potential of pruning on a customized dataset. On condition that the annotation format aligns with that of YOLOv5, it is feasible to implement the pruning on the self-built dataset. If you are still interested in it, maybe you can have a try and let me know if something is still confusing.