Fixed bug in output pkl when batch_size is larger than 1

[LeyangWen]

As described in issue #190, when batch_size is set to larger than 1, the same pkl file will be written multiple times. In the subsequent merge step, the resulting merged file will have frame_no*batch_size number of frames.