Update kth.py

[JunyaoHu]

use torchfile and imageio instead of serialization load_lua to resolve issues that the new version does not support