python - 获取 IndexError:只有整数、切片 (`:`)、省略号 (`...`)、numpy.newaxis (`None`) 和整数或布尔数组是有效的索引
问题描述
我正在尝试使用自定义生成器在 keras tensorflow 中实现多输入模型,如下所示在接受的答案中在 keras 中创建混合数据生成器(图像,csv) :
import random
import pandas as pd
import numpy as np
from glob import glob
from keras.preprocessing import image as krs_image
# Create the arguments for image preprocessing
data_gen_args = dict(
horizontal_flip=True,
brightness_range=[0.5, 1.5],
shear_range=10,
channel_shift_range=50,
rescale=1. / 255,
)
# Create an empty data generator
datagen = ImageDataGenerator()
# Read the image list and csv
image_file_list = glob(f'{images_dir}/{split}/**/*.JPG', recursive=True)
df = pd.read_csv(f'{csv_dir}/{split}.csv', index_col=csv_data[0])
random.shuffle(image_file_list)
def custom_generator(images_list, dataframe, batch_size):
i = 0
while True:
batch = {'images': [], 'csv': [], 'labels': []}
for b in range(batch_size):
if i == len(images_list):
i = 0
random.shuffle(images_list)
# Read image from list and convert to array
image_path = images_list[i]
image_name = os.path.basename(image_path).replace('.JPG', '')
image = krs_image.load_img(image_path, target_size=(img_height, img_width))
image = datagen.apply_transform(image, data_gen_args)
image = krs_image.img_to_array(image)
# Read data from csv using the name of current image
csv_row = dataframe.loc[image_name, :]
label = csv_row['class']
csv_features = csv_row.drop(labels='class')
batch['images'].append(image)
batch['csv'].append(csv_features)
batch['labels'].append(label)
i += 1
batch['images'] = np.array(batch['images'])
batch['csv'] = np.array(batch['csv'])
# Convert labels to categorical values
batch['labels'] = np.eye(num_classes)[batch['labels']]
yield [batch['images'], batch['csv']], batch['labels']
但是,我收到以下索引错误。任何帮助是极大的赞赏。
解决方案
我认为您正在尝试运行block by block and try to running again previously executed block
。另外,这段代码没有问题。将整个代码放在一个块中并再次运行(或重新启动内核并立即运行)。如果通过这种方式你的问题没有解决,你可以试试我的共享脚本。您可以在这里复制并运行我的Colab 文件,希望可以解决您的问题。
推荐阅读
- python - 大学服务器上的 Abaqus 脚本
- android - Android 仪器测试 - 使用 OkHttp MockServer.useHttps(...) 时出现 SSLHandshakeException
- xamarin.forms - 将标签放置在框架内时,标签内容的运行时更改无法正确调整视图大小?
- c++ - 在 C++ 中,什么是 wx.h?
- php - WordPress:从作者那里获取条款
- maven - 测试因未解决的依赖项错误而失败
- json - 如何在 JSON Javascript 中引用兄弟元素?
- php - 数组循环问题,它只显示一次
- java - 在openjdk中使用arthas时找不到java home下的tools.jar
- regex - 正则表达式匹配不带逗号、结束线和空格的文本