import time
import torch.utils.data.distributed
from distributed.protocol import serialize, deserialize
import cv2
import torch
import torch.nn.functional as f
if __name__ == '__main__':
for i in range(1000):
x = torch.rand(100, 3, 352, 352)
# torch.set_num_threads(3)
start = time.time()
torch.save(x, 'd:/img/'+str(i)+'.dat')
aaa= torch.load('d:/img/'+str(i)+'.dat')
print('time4', time.time() - start) # 需要8ms左右。
100张图片,读写一共需要0.8s。
读取需要7ms左右
import time
import torch
import torch.nn.functional as f
if __name__ == '