我正在构建一个具有多个输入的模型,如pyimagesearch所示,但是我无法将所有图像加载到 RAM 中,并且我正在尝试创建一个生成器,该生成器使用flow_from_directory并从 CSV 文件中获取正在处理的每个图像的所有额外属性。
问题:如何从 CSV 中获取与图像生成器中每批图像对应的属性?
def get_combined_generator(images_dir, csv_dir, split, *args):
"""
Creates train/val generators on images and csv data.
Arguments:
images_dir : string
Path to a directory with subdirectories for each class.
csv_dir : string
Path to a directory containing train/val csv files with extra attributes.
split : string
Current split being used (train, val or test)
"""
img_width, img_height, batch_size = args
datagen = ImageDataGenerator(
rescale=1. / 255)
generator = datagen.flow_from_directory( …Run Code Online (Sandbox Code Playgroud) 我正在尝试使用 html5lib 在具有可用列名的一列中获取 50 个状态缩写。
import pandas as pd
import html5lib
fiddy_states = pd.read_html('https://en.wikipedia.org/wiki/List_of_U.S._state_abbreviations')
fs = fiddy_states[0]
Run Code Online (Sandbox Code Playgroud)
在这里我无法更改列名:
fs.rename(columns={4:'abbv'},inplace=True)
Run Code Online (Sandbox Code Playgroud)
在这里,我只能得到我想要的行,但我仍在整理如何在没有可用列名的情况下获得第 4 列:
fs[11:]
Run Code Online (Sandbox Code Playgroud)