Pruning based on self built datasets

HankYe / PAGCP

PAGCP for the compression of YOLOv5

GNU General Public License v3.0

111 stars 11 forks source link

Pruning based on self built datasets #17

Open rememberBr opened 9 months ago

rememberBr commented 9 months ago

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 commented 8 months ago

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.