python 图片划分

import os
import cv2
from random import choice

ImgPath = "E:/Desktop/picture/" 
nameList = os.listdir(ImgPath)

rootPath = "E:/Desktop/person"

count = 787
for i in range(4):    
    for j in range(count):
        temp = choice(nameList)
        
        readPath = ImgPath + temp
        savePath = rootPath + str(i+1) + "/" + temp
        
        img = cv2.imread(readPath)
        cv2.imwrite(savePath, img)
        
        nameList.remove(temp)
    
print(nameList)
print("end !")