- 博客(3)
- 收藏
- 关注
原创 Pytorch计算数据集的均值和标准差。
img = np.array(Image.open(img_path).convert('RGB')) / 255.0 # 准换为RGB的array形式。result = ThreadPool(NUM_THREADS).imap(calc_channel_sum, img_f) # 多线程计算。def calc_channel_sum(img_path): # 计算均值的辅助函数,统计单张图像颜色通道和,以及像素数量。@param data_format: 图片格式(默认为png)
2023-11-01 21:26:15 84
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人