coco-zipf dataset save issue

Hi, 

I appreciate your efforts for the repository and the work.

I ran the repository successfully, everything is running perfectly, the stats are displayed on the terminal, and the dataset coco-zipf created succesfully with .pkl files. I want the images and their respective labels to use them separately.

Kind regards,
