Я новичок в Python, и я написал код для чтения некоторых изображений в 2 for Loops и вычисления некоторых значений, а затем добавил их в словарь, но из-за огромного количества изображений это вообще неэффективно. Я хочу помочь мне для векторизации этого кода. Спасибо
Testimages = [f for f in glob(f'{test_dir}/*.jpg')]
MinDist = float('inf')
test_dictionary = dict()
for img_tst in tqdm(Testimages):
imgTensor_test = ImageLoader(img_tst)
imgTensor_test = to_var(imgTensor_test)
embed_vector_test = model.cnn(imgTensor_test)
for img_fname in tqdm(images):
basename = os.path.basename(img_fname)
foldername = basename.split('.')[-2]
dist = float(np.linalg.norm(embed_vector_test.data.cpu().numpy() - load_array('datalab/whale/train_vector/'+foldername)))
if dist < MinDist :
MinDist = dist
label = img_fname.split('/')[-2] //end second loop
basename_test = os.path.basename(img_tst)
test_dictionary[basename_test]=label //end first loop