求助：创建hdf5训练文件时出错

```Exception in thread Thread-6:
Traceback (most recent call last):
  File "/home/sun/anaconda2/lib/python2.7/threading.py", line 801, in __bootstrap_inner
    self.run()
  File "/home/sun/anaconda2/lib/python2.7/threading.py", line 754, in run
    self.__target(*self.__args, **self.__kwargs)
  File "scripts/make_style_dataset.py", line 73, in write_worker
    imgs_dset[idx] = img.transpose(2, 0, 1)
  File "h5py/_objects.pyx", line 54, in h5py._objects.with_phil.wrapper (/tmp/pip-4rPeHA-build/h5py/_objects.c:2684)
  File "h5py/_objects.pyx", line 55, in h5py._objects.with_phil.wrapper (/tmp/pip-4rPeHA-build/h5py/_objects.c:2642)
  File "/home/sun/.local/lib/python2.7/site-packages/h5py/_hl/dataset.py", line 617, in __setitem__
    for fspace in selection.broadcast(mshape):
  File "/home/sun/.local/lib/python2.7/site-packages/h5py/_hl/selections.py", line 297, in broadcast
    raise TypeError("Can't broadcast %s -> %s" % (target_shape, count))
TypeError: Can't broadcast (4, 256, 256) -> (1, 3, 256, 256)