torch_frame.config.ImageEmbedder
- class ImageEmbedder(*args, **kwargs)[来源]
基类:
ABCimage_embedder的父类,用于ImageEmbedderConfig。该类首先根据数据框中存储的给定路径检索图像,然后将检索到的图像嵌入到张量中。用户负责实现forward_embed(),该方法接收图像列表并返回嵌入张量。用户还可以重写forward_retrieve(),该方法接收图像路径并返回PIL.Image.Image列表。