glob

author: junjie.jiang

Desription

Wrapper of python glob.glob

The parameters are consistent with it.

Code Example

Example

from towhee import DataLoader, pipe, ops
p = (
    pipe.input('image_path')
    .map('image_path', 'image', ops.image_decode.cv2())
    .map('image', 'vec', ops.image_embedding.timm(model_name='resnet50'))
    .output('vec')

)

for data in DataLoader(ops.data_source.glob('./*.jpg')):
    print(p(data).to_list(kv_format=True))

# batch
for data in DataLoader(ops.data_source.glob('./*.jpg'), batch_size=10):
    p.batch(data)