遍历输出当前路径下的npy文件
遍历输出当前路径下的npy文件
import numpy as np
import os
import sys
path = os.getcwd()
dirs = os.listdir(path)
for i in dirs:
if os.path.splitext(i)[1] == ".npy":
pre_train = np.load(i, allow_pickle=True, encoding="latin1")
print("------name-----
原创
2021-04-18 10:57:25 ·
451 阅读 ·
0 评论