Как создать один файл npy для 61 видео и соответствующих им фреймов

Question

У меня 61 видеофайл и соответствующие им кадры. Я извлекаю функции изображения из vgg-16 для каждого видео. Таким образом, выходные формы (94,7,7,512), (111,7,7,512) .... и т. Д. Затем я создаю файл npy для каждого видео. Отсюда у меня 61 файл npy (None,7,7,512).

Но я хочу создать один файл npy для всех видео (61, None, 7,7,512).

Код ниже работает, но не дает мне то, что я хочу

listing = os.listdir("Datasets/CPSM/CPSM_images/")

for file in listing:
    listing_2  = os.listdir("Datasets/CPSM/CPSM_images/" + file + "/" )

    X = []
    for images in listing_2:
        image =  plt.imread("Datasets/CPSM/CPSM_images/" + file + "/" + images )
        X.append (image)
    X = np.array(X)
    print(X.shape)

    image_size=224,
    base_model = VGG16(weights='imagenet', include_top=False, input_shape=(X.shape[1:]))

    batch_size = 48
    XX = base_model.predict(X, batch_size=batch_size, verbose=0, steps=None)
    print(XX.shape)

    B=np.copy(XX)
    np.column_stack((B, XX))
    B=XX

np.save(open("X" + ".npy", 'w'), B)

Я хочу создать один файл npy, который содержит все кадры относительно видео, а именно размер файла npy должен быть (61, None, 224,224,3).

Это возможно?

Answer 1

Я решил это с помощью np.append в цикле в начале кода определения

NP = []

...... ......

в конце цикла добавить это

NP.append(XX)    
np.shape(NP)    
print(np.shape(NP))   

NP = np.array (NP)
np.save (open ("NP.npy", 'w'), NP)