Почему вывод в Google Colab ничего не показывает, тогда как на персональном компьютере он показывает вывод?

O /p показывает ((0,), (0,), (0,)) в Google Colab, а в анаконде он показывает ((296, 150, 150, 3), (296,), (296,)). Код приведен ниже:

import cv2
import glob
    
train_fruit_images = []
train_fruit_labels = [] 
for directory_path in glob.glob('/content/drive/My Drive/Colab Notebooks/Grape/train/Black_Rot/'):
    fruit_label = directory_path.split("\\")[-1]
    for img_path in glob.glob(os.path.join(directory_path, "*.jpg")):
        img = cv2.imread(img_path, cv2.IMREAD_COLOR)       
        img = cv2.resize(img, (224,224)) 
        img = cv2.cvtColor(img, cv2.COLOR_RGB2BGR)
        train_fruit_images.append(img)
        train_fruit_labels.append(fruit_label)
    
    train_fruit_images = np.array(train_fruit_images)
    train_fruit_labels = np.array(train_fruit_labels)
        
        label_to_id = {v:i for i,v in enumerate(np.unique(train_fruit_labels))}
        id_to_label = {v: k for k, v in label_to_id.items()}
    

train_label_ids = np.array([label_to_id[x] for x in train_fruit_labels])
    train_fruit_images.shape, train_label_ids.shape, train_fruit_labels.shape

0 ответов

Другие вопросы по тегам