Image-Text Semantic Matching with AutoMM

Open In Colab Open In SageMaker Studio Lab

Vision and language are two important aspects of human intelligence to understand the real world. Image-text semantic matching, measuring the visual-semantic similarity between image and text, plays a critical role in bridging the vision and language. Learning a joint space where text and image feature vectors are aligned is a typical solution for image-text matching. It is becoming increasingly significant for various vision-and-language tasks, such as cross-modal retrieval, image captioning, text-to-image synthesis, and multimodal neural machine translation. This tutorial will introduce how to apply AutoMM to the image-text matching task.

import os
import warnings
from IPython.display import Image, display
import numpy as np
warnings.filterwarnings('ignore')
np.random.seed(123)

Dataset

In this tutorial, we will use the Flickr30K dataset to demonstrate the image-text matching. The Flickr30k dataset is a popular benchmark for sentence-based picture portrayal. The dataset is comprised of 31,783 images that capture people engaged in everyday activities and events. Each image has a descriptive caption. We organized the dataset using pandas dataframe. To get started, Let’s download the dataset.

from autogluon.core.utils.loaders import load_pd
import pandas as pd
download_dir = './ag_automm_tutorial_imgtxt'
zip_file = 'https://automl-mm-bench.s3.amazonaws.com/flickr30k.zip'
from autogluon.core.utils.loaders import load_zip
load_zip.unzip(zip_file, unzip_dir=download_dir)
Downloading ./ag_automm_tutorial_imgtxt/file.zip from https://automl-mm-bench.s3.amazonaws.com/flickr30k.zip...
0%|          | 0.00/4.38G [00:00<?, ?iB/s]
0%|          | 6.63M/4.38G [00:00<01:22, 53.1MiB/s]
0%|          | 11.9M/4.38G [00:00<02:26, 29.8MiB/s]
0%|          | 15.4M/4.38G [00:00<03:13, 22.6MiB/s]
0%|          | 18.0M/4.38G [00:00<03:52, 18.8MiB/s]
1%|          | 25.2M/4.38G [00:01<03:14, 22.4MiB/s]
1%|          | 33.5M/4.38G [00:01<02:41, 27.0MiB/s]
1%|          | 40.2M/4.38G [00:01<02:22, 30.5MiB/s]
1%|          | 43.4M/4.38G [00:01<02:41, 26.9MiB/s]
1%|          | 49.8M/4.38G [00:01<02:08, 33.8MiB/s]
1%|          | 53.6M/4.38G [00:01<02:30, 28.8MiB/s]
1%|▏         | 57.0M/4.38G [00:02<03:48, 18.9MiB/s]
1%|▏         | 59.5M/4.38G [00:02<03:37, 19.9MiB/s]
2%|▏         | 67.1M/4.38G [00:02<02:24, 29.9MiB/s]
2%|▏         | 73.7M/4.38G [00:02<02:09, 33.2MiB/s]
2%|▏         | 77.8M/4.38G [00:02<02:33, 28.0MiB/s]
2%|▏         | 82.5M/4.38G [00:03<02:33, 28.1MiB/s]
2%|▏         | 85.8M/4.38G [00:03<02:39, 27.0MiB/s]
2%|▏         | 92.3M/4.38G [00:03<02:46, 25.7MiB/s]
2%|▏         | 98.9M/4.38G [00:03<03:03, 23.4MiB/s]
2%|▏         | 101M/4.38G [00:03<03:10, 22.5MiB/s]
2%|▏         | 108M/4.38G [00:04<02:25, 29.3MiB/s]
3%|▎         | 111M/4.38G [00:04<02:35, 27.5MiB/s]
3%|▎         | 117M/4.38G [00:04<02:49, 25.1MiB/s]
3%|▎         | 125M/4.38G [00:04<02:06, 33.6MiB/s]
3%|▎         | 129M/4.38G [00:04<02:28, 28.5MiB/s]
3%|▎         | 133M/4.38G [00:05<03:01, 23.3MiB/s]
3%|▎         | 135M/4.38G [00:05<03:06, 22.8MiB/s]
3%|▎         | 141M/4.38G [00:05<03:06, 22.7MiB/s]
3%|▎         | 143M/4.38G [00:05<03:22, 21.0MiB/s]
3%|▎         | 146M/4.38G [00:05<03:48, 18.5MiB/s]
3%|▎         | 149M/4.38G [00:06<04:01, 17.5MiB/s]
3%|▎         | 151M/4.38G [00:06<04:28, 15.8MiB/s]
4%|▎         | 158M/4.38G [00:06<03:48, 18.5MiB/s]
4%|▎         | 159M/4.38G [00:06<03:49, 18.4MiB/s]
4%|▍         | 166M/4.38G [00:06<03:49, 18.3MiB/s]
4%|▍         | 168M/4.38G [00:07<04:19, 16.2MiB/s]
4%|▍         | 174M/4.38G [00:07<02:57, 23.7MiB/s]
4%|▍         | 177M/4.38G [00:07<03:32, 19.8MiB/s]
4%|▍         | 180M/4.38G [00:07<03:55, 17.9MiB/s]
4%|▍         | 183M/4.38G [00:07<03:44, 18.7MiB/s]
4%|▍         | 185M/4.38G [00:07<04:17, 16.3MiB/s]
4%|▍         | 191M/4.38G [00:08<03:00, 23.2MiB/s]
4%|▍         | 194M/4.38G [00:08<03:23, 20.6MiB/s]
5%|▍         | 200M/4.38G [00:08<04:03, 17.2MiB/s]
5%|▍         | 202M/4.38G [00:08<04:08, 16.8MiB/s]
5%|▍         | 208M/4.38G [00:08<02:52, 24.2MiB/s]
5%|▍         | 211M/4.38G [00:09<02:56, 23.6MiB/s]
5%|▍         | 216M/4.38G [00:09<02:28, 28.0MiB/s]
5%|▌         | 220M/4.38G [00:09<02:50, 24.3MiB/s]
5%|▌         | 223M/4.38G [00:09<02:57, 23.4MiB/s]
5%|▌         | 226M/4.38G [00:09<02:59, 23.1MiB/s]
5%|▌         | 233M/4.38G [00:10<03:26, 20.1MiB/s]
5%|▌         | 235M/4.38G [00:10<03:41, 18.7MiB/s]
6%|▌         | 242M/4.38G [00:10<04:02, 17.1MiB/s]
6%|▌         | 244M/4.38G [00:10<04:16, 16.2MiB/s]
6%|▌         | 250M/4.38G [00:11<03:25, 20.1MiB/s]
6%|▌         | 252M/4.38G [00:11<04:00, 17.1MiB/s]
6%|▌         | 258M/4.38G [00:11<03:29, 19.7MiB/s]
6%|▌         | 260M/4.38G [00:11<03:37, 18.9MiB/s]
6%|▌         | 262M/4.38G [00:11<03:38, 18.9MiB/s]
6%|▌         | 264M/4.38G [00:11<03:34, 19.2MiB/s]
6%|▌         | 267M/4.38G [00:12<03:46, 18.2MiB/s]
6%|▌         | 268M/4.38G [00:12<04:15, 16.1MiB/s]
6%|▋         | 275M/4.38G [00:12<02:38, 25.9MiB/s]
6%|▋         | 278M/4.38G [00:12<02:47, 24.5MiB/s]
7%|▋         | 285M/4.38G [00:12<02:38, 25.9MiB/s]
7%|▋         | 288M/4.38G [00:12<02:54, 23.4MiB/s]
7%|▋         | 294M/4.38G [00:13<02:41, 25.3MiB/s]
7%|▋         | 300M/4.38G [00:13<02:39, 25.7MiB/s]
7%|▋         | 303M/4.38G [00:13<03:24, 19.9MiB/s]
7%|▋         | 310M/4.38G [00:13<02:40, 25.3MiB/s]
7%|▋         | 317M/4.38G [00:13<02:18, 29.4MiB/s]
7%|▋         | 320M/4.38G [00:14<02:36, 26.0MiB/s]
7%|▋         | 325M/4.38G [00:14<02:41, 25.0MiB/s]
7%|▋         | 328M/4.38G [00:14<03:00, 22.5MiB/s]
8%|▊         | 336M/4.38G [00:14<03:00, 22.4MiB/s]
8%|▊         | 344M/4.38G [00:15<02:47, 24.2MiB/s]
8%|▊         | 351M/4.38G [00:15<02:26, 27.6MiB/s]
8%|▊         | 353M/4.38G [00:15<03:05, 21.7MiB/s]
8%|▊         | 361M/4.38G [00:15<02:17, 29.3MiB/s]
8%|▊         | 364M/4.38G [00:15<02:32, 26.4MiB/s]
8%|▊         | 369M/4.38G [00:16<02:56, 22.7MiB/s]
9%|▊         | 376M/4.38G [00:16<02:33, 26.1MiB/s]
9%|▊         | 379M/4.38G [00:16<02:50, 23.5MiB/s]
9%|▉         | 384M/4.38G [00:16<02:54, 23.0MiB/s]
9%|▉         | 387M/4.38G [00:16<03:20, 19.9MiB/s]
9%|▉         | 393M/4.38G [00:17<02:21, 28.2MiB/s]
9%|▉         | 397M/4.38G [00:17<02:28, 26.8MiB/s]
9%|▉         | 401M/4.38G [00:17<02:15, 29.4MiB/s]
9%|▉         | 404M/4.38G [00:17<02:19, 28.5MiB/s]
9%|▉         | 409M/4.38G [00:17<02:46, 23.8MiB/s]
9%|▉         | 412M/4.38G [00:17<03:23, 19.5MiB/s]
10%|▉         | 418M/4.38G [00:18<02:50, 23.3MiB/s]
10%|▉         | 420M/4.38G [00:18<03:00, 21.9MiB/s]
10%|▉         | 426M/4.38G [00:18<02:30, 26.3MiB/s]
10%|▉         | 429M/4.38G [00:18<02:45, 23.9MiB/s]
10%|▉         | 432M/4.38G [00:18<02:38, 25.0MiB/s]
10%|▉         | 434M/4.38G [00:18<02:54, 22.6MiB/s]
10%|▉         | 437M/4.38G [00:19<04:25, 14.8MiB/s]
10%|█         | 443M/4.38G [00:19<03:03, 21.4MiB/s]
10%|█         | 445M/4.38G [00:19<03:11, 20.6MiB/s]
10%|█         | 451M/4.38G [00:19<02:43, 24.0MiB/s]
10%|█         | 454M/4.38G [00:19<03:05, 21.2MiB/s]
10%|█         | 460M/4.38G [00:19<02:29, 26.3MiB/s]
11%|█         | 462M/4.38G [00:20<03:09, 20.6MiB/s]
11%|█         | 468M/4.38G [00:20<02:39, 24.6MiB/s]
11%|█         | 471M/4.38G [00:20<02:59, 21.8MiB/s]
11%|█         | 476M/4.38G [00:20<02:20, 27.8MiB/s]
11%|█         | 480M/4.38G [00:20<02:45, 23.5MiB/s]
11%|█         | 487M/4.38G [00:20<02:08, 30.2MiB/s]
11%|█▏        | 493M/4.38G [00:21<01:48, 35.8MiB/s]
11%|█▏        | 497M/4.38G [00:21<02:13, 29.2MiB/s]
11%|█▏        | 503M/4.38G [00:21<02:04, 31.1MiB/s]
12%|█▏        | 510M/4.38G [00:21<01:50, 34.9MiB/s]
12%|█▏        | 514M/4.38G [00:21<01:53, 34.1MiB/s]
12%|█▏        | 519M/4.38G [00:21<01:53, 34.1MiB/s]
12%|█▏        | 522M/4.38G [00:22<02:24, 26.6MiB/s]
12%|█▏        | 527M/4.38G [00:22<02:13, 28.9MiB/s]
12%|█▏        | 530M/4.38G [00:22<02:34, 24.9MiB/s]
12%|█▏        | 533M/4.38G [00:22<03:20, 19.2MiB/s]
12%|█▏        | 535M/4.38G [00:22<03:54, 16.4MiB/s]
12%|█▏        | 537M/4.38G [00:23<04:36, 13.9MiB/s]
12%|█▏        | 544M/4.38G [00:23<03:27, 18.5MiB/s]
12%|█▏        | 545M/4.38G [00:23<03:33, 17.9MiB/s]
13%|█▎        | 552M/4.38G [00:23<03:07, 20.4MiB/s]
13%|█▎        | 554M/4.38G [00:23<03:48, 16.8MiB/s]
13%|█▎        | 560M/4.38G [00:24<03:08, 20.3MiB/s]
13%|█▎        | 562M/4.38G [00:24<03:28, 18.3MiB/s]
13%|█▎        | 569M/4.38G [00:24<02:49, 22.5MiB/s]
13%|█▎        | 571M/4.38G [00:24<03:21, 18.9MiB/s]
13%|█▎        | 578M/4.38G [00:24<02:15, 28.0MiB/s]
13%|█▎        | 581M/4.38G [00:25<02:54, 21.8MiB/s]
13%|█▎        | 587M/4.38G [00:25<02:50, 22.3MiB/s]
14%|█▎        | 594M/4.38G [00:25<02:28, 25.5MiB/s]
14%|█▎        | 597M/4.38G [00:25<02:49, 22.3MiB/s]
14%|█▍        | 603M/4.38G [00:25<02:11, 28.8MiB/s]
14%|█▍        | 606M/4.38G [00:26<02:27, 25.7MiB/s]
14%|█▍        | 612M/4.38G [00:26<02:09, 29.1MiB/s]
14%|█▍        | 619M/4.38G [00:26<02:40, 23.5MiB/s]
14%|█▍        | 622M/4.38G [00:26<02:43, 23.1MiB/s]
14%|█▍        | 627M/4.38G [00:26<02:15, 27.7MiB/s]
14%|█▍        | 631M/4.38G [00:27<02:36, 24.0MiB/s]
15%|█▍        | 636M/4.38G [00:27<02:14, 27.9MiB/s]
15%|█▍        | 639M/4.38G [00:27<02:27, 25.4MiB/s]
15%|█▍        | 644M/4.38G [00:27<02:29, 25.0MiB/s]
15%|█▍        | 647M/4.38G [00:27<02:45, 22.5MiB/s]
15%|█▍        | 653M/4.38G [00:27<02:17, 27.1MiB/s]
15%|█▍        | 655M/4.38G [00:28<02:51, 21.8MiB/s]
15%|█▌        | 661M/4.38G [00:28<02:37, 23.7MiB/s]
15%|█▌        | 663M/4.38G [00:28<02:40, 23.1MiB/s]
15%|█▌        | 669M/4.38G [00:28<02:36, 23.8MiB/s]
15%|█▌        | 672M/4.38G [00:28<03:18, 18.7MiB/s]
15%|█▌        | 678M/4.38G [00:29<03:52, 15.9MiB/s]
16%|█▌        | 680M/4.38G [00:29<04:06, 15.0MiB/s]
16%|█▌        | 686M/4.38G [00:29<02:50, 21.7MiB/s]
16%|█▌        | 689M/4.38G [00:29<02:44, 22.4MiB/s]
16%|█▌        | 696M/4.38G [00:29<02:04, 29.7MiB/s]
16%|█▌        | 700M/4.38G [00:30<02:43, 22.5MiB/s]
16%|█▌        | 703M/4.38G [00:30<03:01, 20.2MiB/s]
16%|█▌        | 705M/4.38G [00:30<03:16, 18.7MiB/s]
16%|█▌        | 711M/4.38G [00:30<02:43, 22.4MiB/s]
16%|█▋        | 714M/4.38G [00:30<03:04, 19.9MiB/s]
16%|█▋        | 720M/4.38G [00:31<03:02, 20.0MiB/s]
17%|█▋        | 723M/4.38G [00:31<02:44, 22.2MiB/s]
17%|█▋        | 730M/4.38G [00:31<02:33, 23.7MiB/s]
17%|█▋        | 736M/4.38G [00:31<02:09, 28.2MiB/s]
17%|█▋        | 739M/4.38G [00:32<02:45, 22.1MiB/s]
17%|█▋        | 745M/4.38G [00:32<02:59, 20.3MiB/s]
17%|█▋        | 747M/4.38G [00:32<03:00, 20.2MiB/s]
17%|█▋        | 753M/4.38G [00:32<02:33, 23.6MiB/s]
17%|█▋        | 756M/4.38G [00:32<03:13, 18.7MiB/s]
17%|█▋        | 762M/4.38G [00:33<02:27, 24.6MiB/s]
17%|█▋        | 764M/4.38G [00:33<02:47, 21.6MiB/s]
18%|█▊        | 770M/4.38G [00:33<02:43, 22.2MiB/s]
18%|█▊        | 772M/4.38G [00:33<02:56, 20.5MiB/s]
18%|█▊        | 775M/4.38G [00:33<03:01, 19.8MiB/s]
18%|█▊        | 780M/4.38G [00:34<03:02, 19.7MiB/s]
18%|█▊        | 787M/4.38G [00:34<03:06, 19.3MiB/s]
18%|█▊        | 789M/4.38G [00:34<03:37, 16.5MiB/s]
18%|█▊        | 795M/4.38G [00:34<02:51, 20.9MiB/s]
18%|█▊        | 797M/4.38G [00:34<03:17, 18.2MiB/s]
18%|█▊        | 804M/4.38G [00:35<02:14, 26.6MiB/s]
18%|█▊        | 808M/4.38G [00:35<02:23, 25.0MiB/s]
19%|█▊        | 812M/4.38G [00:35<02:14, 26.6MiB/s]
19%|█▊        | 815M/4.38G [00:35<02:23, 24.8MiB/s]
19%|█▊        | 821M/4.38G [00:35<02:22, 25.1MiB/s]
19%|█▉        | 823M/4.38G [00:35<02:46, 21.4MiB/s]
19%|█▉        | 829M/4.38G [00:36<02:21, 25.2MiB/s]
19%|█▉        | 831M/4.38G [00:36<02:53, 20.4MiB/s]
19%|█▉        | 837M/4.38G [00:36<02:20, 25.2MiB/s]
19%|█▉        | 840M/4.38G [00:36<02:15, 26.1MiB/s]
19%|█▉        | 846M/4.38G [00:36<01:44, 33.8MiB/s]
19%|█▉        | 850M/4.38G [00:36<02:11, 26.8MiB/s]
20%|█▉        | 856M/4.38G [00:37<02:15, 26.0MiB/s]
20%|█▉        | 863M/4.38G [00:37<01:42, 34.2MiB/s]
20%|█▉        | 867M/4.38G [00:37<02:08, 27.4MiB/s]
20%|█▉        | 871M/4.38G [00:37<02:12, 26.6MiB/s]
20%|█▉        | 874M/4.38G [00:37<02:37, 22.2MiB/s]
20%|██        | 877M/4.38G [00:37<02:40, 21.8MiB/s]
20%|██        | 879M/4.38G [00:38<02:50, 20.5MiB/s]
20%|██        | 881M/4.38G [00:38<03:32, 16.5MiB/s]
20%|██        | 887M/4.38G [00:38<02:54, 20.0MiB/s]
20%|██        | 890M/4.38G [00:38<04:18, 13.5MiB/s]
20%|██        | 898M/4.38G [00:39<02:55, 19.9MiB/s]
21%|██        | 904M/4.38G [00:39<02:34, 22.5MiB/s]
21%|██        | 907M/4.38G [00:39<02:47, 20.7MiB/s]
21%|██        | 913M/4.38G [00:39<02:40, 21.6MiB/s]
21%|██        | 915M/4.38G [00:39<02:55, 19.8MiB/s]
21%|██        | 921M/4.38G [00:40<02:18, 25.1MiB/s]
21%|██        | 924M/4.38G [00:40<03:05, 18.6MiB/s]
21%|██        | 929M/4.38G [00:40<02:20, 24.5MiB/s]
21%|██▏       | 932M/4.38G [00:40<02:46, 20.7MiB/s]
21%|██▏       | 938M/4.38G [00:41<02:49, 20.3MiB/s]
21%|██▏       | 940M/4.38G [00:41<03:06, 18.5MiB/s]
22%|██▏       | 946M/4.38G [00:41<02:41, 21.3MiB/s]
22%|██▏       | 948M/4.38G [00:41<02:53, 19.7MiB/s]
22%|██▏       | 955M/4.38G [00:41<02:06, 27.2MiB/s]
22%|██▏       | 958M/4.38G [00:41<02:06, 27.1MiB/s]
22%|██▏       | 965M/4.38G [00:41<01:40, 34.0MiB/s]
22%|██▏       | 972M/4.38G [00:42<01:46, 32.1MiB/s]
22%|██▏       | 975M/4.38G [00:42<02:30, 22.6MiB/s]
22%|██▏       | 981M/4.38G [00:42<02:10, 26.1MiB/s]
23%|██▎       | 988M/4.38G [00:42<01:47, 31.5MiB/s]
23%|██▎       | 992M/4.38G [00:43<02:07, 26.5MiB/s]
23%|██▎       | 996M/4.38G [00:43<03:19, 17.0MiB/s]
23%|██▎       | 999M/4.38G [00:43<03:25, 16.4MiB/s]
23%|██▎       | 1.00G/4.38G [00:43<02:53, 19.4MiB/s]
23%|██▎       | 1.01G/4.38G [00:44<03:24, 16.5MiB/s]
23%|██▎       | 1.01G/4.38G [00:44<04:20, 13.0MiB/s]
23%|██▎       | 1.01G/4.38G [00:44<04:02, 13.9MiB/s]
23%|██▎       | 1.02G/4.38G [00:44<03:44, 15.0MiB/s]
23%|██▎       | 1.02G/4.38G [00:44<02:38, 21.2MiB/s]
23%|██▎       | 1.02G/4.38G [00:45<02:51, 19.6MiB/s]
24%|██▎       | 1.03G/4.38G [00:45<02:14, 24.9MiB/s]
24%|██▎       | 1.03G/4.38G [00:45<02:36, 21.4MiB/s]
24%|██▎       | 1.04G/4.38G [00:45<02:47, 20.0MiB/s]
24%|██▍       | 1.04G/4.38G [00:45<02:50, 19.6MiB/s]
24%|██▍       | 1.05G/4.38G [00:46<03:40, 15.1MiB/s]
24%|██▍       | 1.05G/4.38G [00:46<04:01, 13.8MiB/s]
24%|██▍       | 1.06G/4.38G [00:46<03:10, 17.4MiB/s]
24%|██▍       | 1.06G/4.38G [00:47<03:49, 14.5MiB/s]
24%|██▍       | 1.06G/4.38G [00:47<02:29, 22.2MiB/s]
24%|██▍       | 1.07G/4.38G [00:47<02:58, 18.5MiB/s]
24%|██▍       | 1.07G/4.38G [00:47<02:31, 21.8MiB/s]
25%|██▍       | 1.07G/4.38G [00:47<02:36, 21.2MiB/s]
25%|██▍       | 1.08G/4.38G [00:48<02:19, 23.6MiB/s]
25%|██▍       | 1.09G/4.38G [00:48<01:53, 29.0MiB/s]
25%|██▍       | 1.09G/4.38G [00:48<02:09, 25.4MiB/s]
25%|██▌       | 1.10G/4.38G [00:48<01:50, 29.6MiB/s]
25%|██▌       | 1.11G/4.38G [00:48<01:31, 35.7MiB/s]
25%|██▌       | 1.11G/4.38G [00:48<01:25, 38.1MiB/s]
26%|██▌       | 1.12G/4.38G [00:49<01:33, 35.0MiB/s]
26%|██▌       | 1.12G/4.38G [00:49<01:27, 37.3MiB/s]
26%|██▌       | 1.13G/4.38G [00:49<02:09, 25.2MiB/s]
26%|██▌       | 1.13G/4.38G [00:49<01:48, 29.9MiB/s]
26%|██▌       | 1.14G/4.38G [00:49<01:54, 28.3MiB/s]
26%|██▌       | 1.14G/4.38G [00:50<02:13, 24.3MiB/s]
26%|██▌       | 1.15G/4.38G [00:50<02:05, 25.8MiB/s]
26%|██▋       | 1.15G/4.38G [00:50<02:33, 21.1MiB/s]
26%|██▋       | 1.16G/4.38G [00:50<02:17, 23.5MiB/s]
26%|██▋       | 1.16G/4.38G [00:50<02:46, 19.4MiB/s]
27%|██▋       | 1.16G/4.38G [00:51<02:44, 19.5MiB/s]
27%|██▋       | 1.17G/4.38G [00:51<03:01, 17.7MiB/s]
27%|██▋       | 1.17G/4.38G [00:51<02:04, 25.8MiB/s]
27%|██▋       | 1.18G/4.38G [00:51<02:24, 22.2MiB/s]
27%|██▋       | 1.18G/4.38G [00:51<02:13, 23.9MiB/s]
27%|██▋       | 1.19G/4.38G [00:52<01:54, 27.9MiB/s]
27%|██▋       | 1.19G/4.38G [00:52<02:00, 26.4MiB/s]
27%|██▋       | 1.20G/4.38G [00:52<01:50, 28.9MiB/s]
27%|██▋       | 1.20G/4.38G [00:52<01:55, 27.4MiB/s]
28%|██▊       | 1.21G/4.38G [00:52<01:52, 28.2MiB/s]
28%|██▊       | 1.21G/4.38G [00:52<01:33, 33.9MiB/s]
28%|██▊       | 1.22G/4.38G [00:53<01:46, 29.6MiB/s]
28%|██▊       | 1.22G/4.38G [00:53<01:52, 28.1MiB/s]
28%|██▊       | 1.22G/4.38G [00:53<02:01, 26.0MiB/s]
28%|██▊       | 1.23G/4.38G [00:53<01:35, 33.0MiB/s]
28%|██▊       | 1.23G/4.38G [00:53<01:59, 26.4MiB/s]
28%|██▊       | 1.24G/4.38G [00:53<01:58, 26.5MiB/s]
28%|██▊       | 1.24G/4.38G [00:54<02:21, 22.2MiB/s]
29%|██▊       | 1.25G/4.38G [00:54<01:45, 29.7MiB/s]
29%|██▊       | 1.25G/4.38G [00:54<02:09, 24.2MiB/s]
29%|██▊       | 1.26G/4.38G [00:54<02:05, 25.0MiB/s]
29%|██▉       | 1.26G/4.38G [00:54<01:53, 27.5MiB/s]
29%|██▉       | 1.27G/4.38G [00:54<02:04, 24.9MiB/s]
29%|██▉       | 1.27G/4.38G [00:55<02:01, 25.5MiB/s]
29%|██▉       | 1.28G/4.38G [00:55<02:31, 20.5MiB/s]
29%|██▉       | 1.28G/4.38G [00:55<02:11, 23.6MiB/s]
29%|██▉       | 1.28G/4.38G [00:55<02:19, 22.1MiB/s]
29%|██▉       | 1.29G/4.38G [00:55<01:43, 29.7MiB/s]
30%|██▉       | 1.30G/4.38G [00:56<01:42, 30.0MiB/s]
30%|██▉       | 1.30G/4.38G [00:56<01:48, 28.5MiB/s]
30%|██▉       | 1.31G/4.38G [00:56<02:00, 25.4MiB/s]
30%|██▉       | 1.31G/4.38G [00:56<02:10, 23.5MiB/s]
30%|███       | 1.32G/4.38G [00:56<02:14, 22.7MiB/s]
30%|███       | 1.32G/4.38G [00:57<02:36, 19.6MiB/s]
30%|███       | 1.32G/4.38G [00:57<02:18, 22.1MiB/s]
30%|███       | 1.33G/4.38G [00:57<02:32, 20.0MiB/s]
30%|███       | 1.33G/4.38G [00:57<02:37, 19.4MiB/s]
30%|███       | 1.33G/4.38G [00:57<02:23, 21.2MiB/s]
30%|███       | 1.34G/4.38G [00:57<02:28, 20.5MiB/s]
31%|███       | 1.34G/4.38G [00:58<01:56, 26.2MiB/s]
31%|███       | 1.34G/4.38G [00:58<02:08, 23.7MiB/s]
31%|███       | 1.35G/4.38G [00:58<02:55, 17.3MiB/s]
31%|███       | 1.35G/4.38G [00:58<03:13, 15.7MiB/s]
31%|███       | 1.36G/4.38G [00:59<02:23, 21.1MiB/s]
31%|███       | 1.36G/4.38G [00:59<02:43, 18.5MiB/s]
31%|███       | 1.37G/4.38G [00:59<02:13, 22.6MiB/s]
31%|███       | 1.37G/4.38G [00:59<02:50, 17.7MiB/s]
31%|███▏      | 1.37G/4.38G [00:59<02:54, 17.3MiB/s]
31%|███▏      | 1.38G/4.38G [01:00<03:00, 16.7MiB/s]
32%|███▏      | 1.38G/4.38G [01:00<02:06, 23.8MiB/s]
32%|███▏      | 1.39G/4.38G [01:00<01:32, 32.5MiB/s]
32%|███▏      | 1.40G/4.38G [01:00<01:53, 26.3MiB/s]
32%|███▏      | 1.40G/4.38G [01:00<02:05, 23.7MiB/s]
32%|███▏      | 1.40G/4.38G [01:01<02:09, 22.9MiB/s]
32%|███▏      | 1.41G/4.38G [01:01<02:21, 21.0MiB/s]
32%|███▏      | 1.41G/4.38G [01:01<02:33, 19.4MiB/s]
32%|███▏      | 1.42G/4.38G [01:01<02:02, 24.3MiB/s]
33%|███▎      | 1.42G/4.38G [01:02<02:27, 20.1MiB/s]
33%|███▎      | 1.43G/4.38G [01:02<02:57, 16.6MiB/s]
33%|███▎      | 1.43G/4.38G [01:02<02:37, 18.8MiB/s]
33%|███▎      | 1.43G/4.38G [01:02<02:59, 16.4MiB/s]
33%|███▎      | 1.44G/4.38G [01:03<02:19, 21.1MiB/s]
33%|███▎      | 1.45G/4.38G [01:03<02:24, 20.3MiB/s]
33%|███▎      | 1.45G/4.38G [01:03<02:31, 19.4MiB/s]
33%|███▎      | 1.46G/4.38G [01:03<02:01, 24.0MiB/s]
33%|███▎      | 1.46G/4.38G [01:03<02:10, 22.4MiB/s]
33%|███▎      | 1.47G/4.38G [01:04<01:53, 25.7MiB/s]
34%|███▎      | 1.47G/4.38G [01:04<01:58, 24.6MiB/s]
34%|███▎      | 1.47G/4.38G [01:04<01:46, 27.2MiB/s]
34%|███▎      | 1.48G/4.38G [01:04<02:08, 22.6MiB/s]
34%|███▍      | 1.48G/4.38G [01:04<01:49, 26.4MiB/s]
34%|███▍      | 1.49G/4.38G [01:04<02:00, 24.0MiB/s]
34%|███▍      | 1.49G/4.38G [01:05<01:42, 28.3MiB/s]
34%|███▍      | 1.49G/4.38G [01:05<01:45, 27.4MiB/s]
34%|███▍      | 1.50G/4.38G [01:05<01:45, 27.2MiB/s]
34%|███▍      | 1.50G/4.38G [01:05<01:56, 24.7MiB/s]
34%|███▍      | 1.51G/4.38G [01:05<01:38, 29.2MiB/s]
35%|███▍      | 1.52G/4.38G [01:05<01:25, 33.6MiB/s]
35%|███▍      | 1.52G/4.38G [01:05<01:32, 30.9MiB/s]
35%|███▍      | 1.52G/4.38G [01:06<01:42, 27.8MiB/s]
35%|███▍      | 1.53G/4.38G [01:06<01:58, 24.0MiB/s]
35%|███▍      | 1.53G/4.38G [01:06<02:29, 19.1MiB/s]
35%|███▌      | 1.53G/4.38G [01:06<02:24, 19.7MiB/s]
35%|███▌      | 1.54G/4.38G [01:06<02:25, 19.5MiB/s]
35%|███▌      | 1.54G/4.38G [01:07<02:02, 23.2MiB/s]
35%|███▌      | 1.55G/4.38G [01:07<01:57, 24.1MiB/s]
35%|███▌      | 1.55G/4.38G [01:07<02:12, 21.3MiB/s]
36%|███▌      | 1.56G/4.38G [01:07<02:14, 20.9MiB/s]
36%|███▌      | 1.56G/4.38G [01:08<02:30, 18.8MiB/s]
36%|███▌      | 1.57G/4.38G [01:08<02:29, 18.8MiB/s]
36%|███▌      | 1.57G/4.38G [01:08<02:48, 16.7MiB/s]
36%|███▌      | 1.58G/4.38G [01:08<02:15, 20.7MiB/s]
36%|███▌      | 1.58G/4.38G [01:09<02:39, 17.5MiB/s]
36%|███▌      | 1.58G/4.38G [01:09<02:02, 22.9MiB/s]
36%|███▌      | 1.59G/4.38G [01:09<02:12, 21.0MiB/s]
36%|███▋      | 1.59G/4.38G [01:09<01:44, 26.7MiB/s]
36%|███▋      | 1.60G/4.38G [01:09<02:07, 21.8MiB/s]
37%|███▋      | 1.60G/4.38G [01:09<01:47, 25.8MiB/s]
37%|███▋      | 1.60G/4.38G [01:10<02:00, 23.1MiB/s]
37%|███▋      | 1.61G/4.38G [01:10<01:44, 26.5MiB/s]
37%|███▋      | 1.62G/4.38G [01:10<01:23, 33.2MiB/s]
37%|███▋      | 1.62G/4.38G [01:10<02:33, 18.0MiB/s]
37%|███▋      | 1.63G/4.38G [01:11<02:11, 21.0MiB/s]
37%|███▋      | 1.63G/4.38G [01:11<02:53, 15.8MiB/s]
37%|███▋      | 1.63G/4.38G [01:11<02:11, 20.9MiB/s]
37%|███▋      | 1.64G/4.38G [01:11<02:15, 20.2MiB/s]
38%|███▊      | 1.64G/4.38G [01:11<01:55, 23.6MiB/s]
38%|███▊      | 1.65G/4.38G [01:12<02:04, 22.0MiB/s]
38%|███▊      | 1.65G/4.38G [01:12<02:25, 18.7MiB/s]
38%|███▊      | 1.65G/4.38G [01:12<02:28, 18.4MiB/s]
38%|███▊      | 1.66G/4.38G [01:12<01:33, 29.2MiB/s]
38%|███▊      | 1.66G/4.38G [01:12<01:38, 27.5MiB/s]
38%|███▊      | 1.67G/4.38G [01:13<02:15, 20.0MiB/s]
38%|███▊      | 1.67G/4.38G [01:13<02:42, 16.6MiB/s]
38%|███▊      | 1.68G/4.38G [01:13<01:55, 23.3MiB/s]
38%|███▊      | 1.68G/4.38G [01:13<01:31, 29.5MiB/s]
39%|███▊      | 1.69G/4.38G [01:13<01:37, 27.6MiB/s]
39%|███▊      | 1.69G/4.38G [01:13<01:32, 29.0MiB/s]
39%|███▊      | 1.70G/4.38G [01:14<02:00, 22.3MiB/s]
39%|███▉      | 1.70G/4.38G [01:14<01:59, 22.4MiB/s]
39%|███▉      | 1.70G/4.38G [01:14<02:21, 18.9MiB/s]
39%|███▉      | 1.71G/4.38G [01:14<01:57, 22.7MiB/s]
39%|███▉      | 1.71G/4.38G [01:14<02:24, 18.5MiB/s]
39%|███▉      | 1.72G/4.38G [01:15<01:47, 24.9MiB/s]
39%|███▉      | 1.72G/4.38G [01:15<01:48, 24.6MiB/s]
39%|███▉      | 1.73G/4.38G [01:15<01:38, 26.9MiB/s]
39%|███▉      | 1.73G/4.38G [01:15<02:08, 20.7MiB/s]
40%|███▉      | 1.74G/4.38G [01:15<01:46, 24.9MiB/s]
40%|███▉      | 1.74G/4.38G [01:16<01:37, 27.1MiB/s]
40%|███▉      | 1.75G/4.38G [01:16<01:29, 29.4MiB/s]
40%|████      | 1.75G/4.38G [01:16<01:47, 24.4MiB/s]
40%|████      | 1.76G/4.38G [01:16<01:27, 30.1MiB/s]
40%|████      | 1.77G/4.38G [01:16<01:22, 31.6MiB/s]
40%|████      | 1.77G/4.38G [01:17<01:40, 26.1MiB/s]
41%|████      | 1.78G/4.38G [01:17<01:43, 25.2MiB/s]
41%|████      | 1.78G/4.38G [01:17<01:45, 24.6MiB/s]
41%|████      | 1.79G/4.38G [01:18<02:45, 15.6MiB/s]
41%|████      | 1.79G/4.38G [01:18<03:09, 13.7MiB/s]
41%|████      | 1.79G/4.38G [01:18<02:40, 16.1MiB/s]
41%|████      | 1.80G/4.38G [01:18<02:38, 16.3MiB/s]
41%|████      | 1.80G/4.38G [01:18<01:52, 22.8MiB/s]
41%|████      | 1.80G/4.38G [01:18<01:56, 22.1MiB/s]
41%|████▏     | 1.81G/4.38G [01:19<01:45, 24.4MiB/s]
41%|████▏     | 1.81G/4.38G [01:19<01:59, 21.6MiB/s]
42%|████▏     | 1.82G/4.38G [01:19<01:31, 28.0MiB/s]
42%|████▏     | 1.82G/4.38G [01:19<01:41, 25.2MiB/s]
42%|████▏     | 1.83G/4.38G [01:19<01:43, 24.6MiB/s]
42%|████▏     | 1.83G/4.38G [01:20<02:02, 20.9MiB/s]
42%|████▏     | 1.84G/4.38G [01:20<01:42, 24.8MiB/s]
42%|████▏     | 1.84G/4.38G [01:20<01:58, 21.5MiB/s]
42%|████▏     | 1.84G/4.38G [01:20<01:33, 27.0MiB/s]
42%|████▏     | 1.85G/4.38G [01:20<01:49, 23.2MiB/s]
42%|████▏     | 1.85G/4.38G [01:21<02:57, 14.3MiB/s]
42%|████▏     | 1.85G/4.38G [01:21<03:09, 13.3MiB/s]
43%|████▎     | 1.86G/4.38G [01:21<02:14, 18.8MiB/s]
43%|████▎     | 1.87G/4.38G [01:21<01:45, 23.9MiB/s]
43%|████▎     | 1.87G/4.38G [01:22<01:45, 23.9MiB/s]
43%|████▎     | 1.88G/4.38G [01:22<01:43, 24.1MiB/s]
43%|████▎     | 1.88G/4.38G [01:22<01:50, 22.6MiB/s]
43%|████▎     | 1.88G/4.38G [01:22<01:49, 22.8MiB/s]
43%|████▎     | 1.89G/4.38G [01:22<01:40, 24.7MiB/s]
43%|████▎     | 1.89G/4.38G [01:22<01:45, 23.7MiB/s]
43%|████▎     | 1.89G/4.38G [01:22<01:38, 25.3MiB/s]
43%|████▎     | 1.90G/4.38G [01:23<02:08, 19.4MiB/s]
43%|████▎     | 1.90G/4.38G [01:23<02:08, 19.3MiB/s]
43%|████▎     | 1.90G/4.38G [01:23<02:21, 17.5MiB/s]
44%|████▎     | 1.91G/4.38G [01:23<01:43, 24.0MiB/s]
44%|████▎     | 1.91G/4.38G [01:23<01:55, 21.4MiB/s]
44%|████▍     | 1.92G/4.38G [01:24<01:41, 24.2MiB/s]
44%|████▍     | 1.92G/4.38G [01:24<01:58, 20.8MiB/s]
44%|████▍     | 1.93G/4.38G [01:24<01:33, 26.3MiB/s]
44%|████▍     | 1.93G/4.38G [01:24<01:57, 20.9MiB/s]
44%|████▍     | 1.94G/4.38G [01:24<01:37, 25.0MiB/s]
44%|████▍     | 1.94G/4.38G [01:25<01:21, 30.0MiB/s]
44%|████▍     | 1.95G/4.38G [01:25<01:33, 26.0MiB/s]
45%|████▍     | 1.95G/4.38G [01:25<01:12, 33.7MiB/s]
45%|████▍     | 1.96G/4.38G [01:25<01:28, 27.4MiB/s]
45%|████▍     | 1.96G/4.38G [01:25<01:29, 26.9MiB/s]
45%|████▍     | 1.97G/4.38G [01:25<01:14, 32.5MiB/s]
45%|████▌     | 1.97G/4.38G [01:26<01:24, 28.3MiB/s]
45%|████▌     | 1.98G/4.38G [01:26<01:17, 31.0MiB/s]
45%|████▌     | 1.98G/4.38G [01:26<01:29, 26.8MiB/s]
45%|████▌     | 1.99G/4.38G [01:26<01:54, 21.0MiB/s]
45%|████▌     | 1.99G/4.38G [01:26<02:03, 19.4MiB/s]
46%|████▌     | 2.00G/4.38G [01:27<01:44, 22.8MiB/s]
46%|████▌     | 2.00G/4.38G [01:27<01:52, 21.1MiB/s]
46%|████▌     | 2.00G/4.38G [01:27<01:32, 25.8MiB/s]
46%|████▌     | 2.01G/4.38G [01:27<01:53, 20.8MiB/s]
46%|████▌     | 2.01G/4.38G [01:27<01:48, 21.9MiB/s]
46%|████▌     | 2.02G/4.38G [01:28<01:34, 25.0MiB/s]
46%|████▌     | 2.02G/4.38G [01:28<01:42, 23.0MiB/s]
46%|████▋     | 2.03G/4.38G [01:28<01:39, 23.7MiB/s]
46%|████▋     | 2.03G/4.38G [01:28<01:43, 22.7MiB/s]
47%|████▋     | 2.04G/4.38G [01:28<01:14, 31.3MiB/s]
47%|████▋     | 2.04G/4.38G [01:29<01:33, 25.0MiB/s]
47%|████▋     | 2.05G/4.38G [01:29<01:38, 23.7MiB/s]
47%|████▋     | 2.05G/4.38G [01:29<01:26, 26.8MiB/s]
47%|████▋     | 2.06G/4.38G [01:29<01:38, 23.7MiB/s]
47%|████▋     | 2.06G/4.38G [01:29<01:38, 23.6MiB/s]
47%|████▋     | 2.06G/4.38G [01:30<01:52, 20.6MiB/s]
47%|████▋     | 2.07G/4.38G [01:30<01:38, 23.5MiB/s]
47%|████▋     | 2.07G/4.38G [01:30<02:09, 17.9MiB/s]
47%|████▋     | 2.08G/4.38G [01:30<01:42, 22.5MiB/s]
48%|████▊     | 2.08G/4.38G [01:30<01:54, 20.1MiB/s]
48%|████▊     | 2.09G/4.38G [01:31<01:36, 23.9MiB/s]
48%|████▊     | 2.09G/4.38G [01:31<01:37, 23.6MiB/s]
48%|████▊     | 2.09G/4.38G [01:31<01:44, 21.9MiB/s]
48%|████▊     | 2.10G/4.38G [01:31<01:40, 22.7MiB/s]
48%|████▊     | 2.10G/4.38G [01:31<01:12, 31.3MiB/s]
48%|████▊     | 2.11G/4.38G [01:31<01:18, 28.8MiB/s]
48%|████▊     | 2.11G/4.38G [01:32<01:35, 23.8MiB/s]
48%|████▊     | 2.12G/4.38G [01:32<01:41, 22.4MiB/s]
48%|████▊     | 2.12G/4.38G [01:32<01:37, 23.2MiB/s]
49%|████▊     | 2.13G/4.38G [01:32<01:15, 29.8MiB/s]
49%|████▊     | 2.13G/4.38G [01:32<01:32, 24.3MiB/s]
49%|████▉     | 2.14G/4.38G [01:33<01:30, 24.9MiB/s]
49%|████▉     | 2.15G/4.38G [01:33<01:14, 30.0MiB/s]
49%|████▉     | 2.15G/4.38G [01:33<01:25, 26.1MiB/s]
49%|████▉     | 2.15G/4.38G [01:33<01:37, 22.9MiB/s]
49%|████▉     | 2.16G/4.38G [01:33<01:57, 18.9MiB/s]
49%|████▉     | 2.16G/4.38G [01:34<01:55, 19.2MiB/s]
49%|████▉     | 2.16G/4.38G [01:34<02:21, 15.6MiB/s]
50%|████▉     | 2.17G/4.38G [01:34<01:38, 22.4MiB/s]
50%|████▉     | 2.18G/4.38G [01:35<01:49, 20.0MiB/s]
50%|████▉     | 2.18G/4.38G [01:35<02:00, 18.3MiB/s]
50%|████▉     | 2.19G/4.38G [01:35<01:31, 23.9MiB/s]
50%|█████     | 2.19G/4.38G [01:35<01:30, 24.2MiB/s]
50%|█████     | 2.20G/4.38G [01:35<01:25, 25.7MiB/s]
50%|█████     | 2.20G/4.38G [01:35<01:32, 23.6MiB/s]
50%|█████     | 2.20G/4.38G [01:36<02:00, 18.0MiB/s]
50%|█████     | 2.21G/4.38G [01:36<01:43, 21.0MiB/s]
50%|█████     | 2.21G/4.38G [01:36<01:50, 19.7MiB/s]
51%|█████     | 2.21G/4.38G [01:36<01:31, 23.6MiB/s]
51%|█████     | 2.22G/4.38G [01:36<01:53, 19.1MiB/s]
51%|█████     | 2.22G/4.38G [01:36<01:21, 26.4MiB/s]
51%|█████     | 2.23G/4.38G [01:37<01:22, 26.2MiB/s]
51%|█████     | 2.23G/4.38G [01:37<01:55, 18.7MiB/s]
51%|█████     | 2.24G/4.38G [01:37<01:34, 22.8MiB/s]
51%|█████     | 2.24G/4.38G [01:37<01:42, 21.0MiB/s]
51%|█████▏    | 2.25G/4.38G [01:38<01:32, 23.2MiB/s]
51%|█████▏    | 2.25G/4.38G [01:38<01:33, 22.9MiB/s]
51%|█████▏    | 2.25G/4.38G [01:38<02:49, 12.6MiB/s]
52%|█████▏    | 2.26G/4.38G [01:39<02:52, 12.3MiB/s]
52%|█████▏    | 2.26G/4.38G [01:39<01:50, 19.1MiB/s]
52%|█████▏    | 2.27G/4.38G [01:39<01:30, 23.4MiB/s]
52%|█████▏    | 2.27G/4.38G [01:39<01:45, 19.9MiB/s]
52%|█████▏    | 2.28G/4.38G [01:39<01:28, 23.8MiB/s]
52%|█████▏    | 2.29G/4.38G [01:40<01:12, 28.8MiB/s]
52%|█████▏    | 2.29G/4.38G [01:40<01:35, 21.8MiB/s]
52%|█████▏    | 2.30G/4.38G [01:40<01:22, 25.4MiB/s]
52%|█████▏    | 2.30G/4.38G [01:40<01:44, 19.9MiB/s]
53%|█████▎    | 2.31G/4.38G [01:41<01:39, 20.8MiB/s]
53%|█████▎    | 2.31G/4.38G [01:41<01:50, 18.7MiB/s]
53%|█████▎    | 2.31G/4.38G [01:41<01:27, 23.7MiB/s]
53%|█████▎    | 2.32G/4.38G [01:41<01:44, 19.7MiB/s]
53%|█████▎    | 2.32G/4.38G [01:41<01:50, 18.7MiB/s]
53%|█████▎    | 2.32G/4.38G [01:42<01:55, 17.8MiB/s]
53%|█████▎    | 2.33G/4.38G [01:42<01:22, 24.9MiB/s]
53%|█████▎    | 2.33G/4.38G [01:42<01:34, 21.6MiB/s]
53%|█████▎    | 2.34G/4.38G [01:42<01:42, 20.0MiB/s]
53%|█████▎    | 2.34G/4.38G [01:42<01:51, 18.3MiB/s]
54%|█████▎    | 2.35G/4.38G [01:42<01:11, 28.2MiB/s]
54%|█████▎    | 2.35G/4.38G [01:43<01:13, 27.7MiB/s]
54%|█████▍    | 2.36G/4.38G [01:43<01:16, 26.3MiB/s]
54%|█████▍    | 2.36G/4.38G [01:43<01:10, 28.6MiB/s]
54%|█████▍    | 2.37G/4.38G [01:43<01:25, 23.7MiB/s]
54%|█████▍    | 2.37G/4.38G [01:43<01:16, 26.4MiB/s]
54%|█████▍    | 2.38G/4.38G [01:44<01:31, 22.0MiB/s]
54%|█████▍    | 2.38G/4.38G [01:44<01:13, 27.2MiB/s]
54%|█████▍    | 2.38G/4.38G [01:44<01:30, 22.0MiB/s]
55%|█████▍    | 2.39G/4.38G [01:44<01:39, 20.0MiB/s]
55%|█████▍    | 2.39G/4.38G [01:45<01:55, 17.3MiB/s]
55%|█████▍    | 2.40G/4.38G [01:45<01:35, 20.8MiB/s]
55%|█████▍    | 2.40G/4.38G [01:45<01:49, 18.1MiB/s]
55%|█████▍    | 2.41G/4.38G [01:45<01:31, 21.6MiB/s]
55%|█████▍    | 2.41G/4.38G [01:45<01:52, 17.5MiB/s]
55%|█████▌    | 2.41G/4.38G [01:46<01:46, 18.4MiB/s]
55%|█████▌    | 2.42G/4.38G [01:46<01:52, 17.5MiB/s]
55%|█████▌    | 2.42G/4.38G [01:46<02:09, 15.2MiB/s]
55%|█████▌    | 2.42G/4.38G [01:46<02:00, 16.2MiB/s]
56%|█████▌    | 2.43G/4.38G [01:46<01:16, 25.6MiB/s]
56%|█████▌    | 2.44G/4.38G [01:47<01:23, 23.2MiB/s]
56%|█████▌    | 2.44G/4.38G [01:47<01:21, 23.7MiB/s]
56%|█████▌    | 2.45G/4.38G [01:47<00:59, 32.2MiB/s]
56%|█████▌    | 2.46G/4.38G [01:47<01:07, 28.4MiB/s]
56%|█████▌    | 2.46G/4.38G [01:47<00:59, 32.3MiB/s]
56%|█████▋    | 2.47G/4.38G [01:48<01:22, 23.2MiB/s]
56%|█████▋    | 2.47G/4.38G [01:48<01:09, 27.3MiB/s]
57%|█████▋    | 2.48G/4.38G [01:48<00:59, 31.9MiB/s]
57%|█████▋    | 2.49G/4.38G [01:48<01:04, 29.2MiB/s]
57%|█████▋    | 2.49G/4.38G [01:49<01:08, 27.4MiB/s]
57%|█████▋    | 2.50G/4.38G [01:49<01:38, 19.1MiB/s]
57%|█████▋    | 2.50G/4.38G [01:49<01:35, 19.8MiB/s]
57%|█████▋    | 2.51G/4.38G [01:49<01:18, 24.0MiB/s]
57%|█████▋    | 2.51G/4.38G [01:50<01:27, 21.4MiB/s]
57%|█████▋    | 2.52G/4.38G [01:50<01:11, 26.1MiB/s]
57%|█████▋    | 2.52G/4.38G [01:50<01:22, 22.5MiB/s]
58%|█████▊    | 2.52G/4.38G [01:50<01:26, 21.4MiB/s]
58%|█████▊    | 2.52G/4.38G [01:50<01:33, 19.9MiB/s]
58%|█████▊    | 2.53G/4.38G [01:50<01:06, 27.9MiB/s]
58%|█████▊    | 2.53G/4.38G [01:51<01:15, 24.5MiB/s]
58%|█████▊    | 2.54G/4.38G [01:51<01:17, 23.7MiB/s]
58%|█████▊    | 2.54G/4.38G [01:51<01:17, 23.9MiB/s]
58%|█████▊    | 2.55G/4.38G [01:51<01:13, 24.7MiB/s]
58%|█████▊    | 2.56G/4.38G [01:52<01:19, 23.0MiB/s]
59%|█████▊    | 2.57G/4.38G [01:52<01:25, 21.3MiB/s]
59%|█████▊    | 2.57G/4.38G [01:52<01:35, 19.1MiB/s]
59%|█████▊    | 2.57G/4.38G [01:52<01:13, 24.5MiB/s]
59%|█████▉    | 2.58G/4.38G [01:52<01:20, 22.4MiB/s]
59%|█████▉    | 2.58G/4.38G [01:53<01:19, 22.5MiB/s]
59%|█████▉    | 2.58G/4.38G [01:53<01:23, 21.4MiB/s]
59%|█████▉    | 2.59G/4.38G [01:53<01:52, 16.0MiB/s]
59%|█████▉    | 2.59G/4.38G [01:53<01:33, 19.2MiB/s]
59%|█████▉    | 2.59G/4.38G [01:53<01:44, 17.2MiB/s]
59%|█████▉    | 2.60G/4.38G [01:54<01:26, 20.7MiB/s]
59%|█████▉    | 2.60G/4.38G [01:54<01:29, 19.8MiB/s]
60%|█████▉    | 2.61G/4.38G [01:54<01:00, 29.3MiB/s]
60%|█████▉    | 2.61G/4.38G [01:54<01:16, 23.3MiB/s]
60%|█████▉    | 2.62G/4.38G [01:54<01:14, 23.7MiB/s]
60%|█████▉    | 2.62G/4.38G [01:54<01:15, 23.2MiB/s]
60%|█████▉    | 2.62G/4.38G [01:55<01:14, 23.7MiB/s]
60%|█████▉    | 2.63G/4.38G [01:55<01:23, 21.0MiB/s]
60%|██████    | 2.63G/4.38G [01:55<01:07, 25.9MiB/s]
60%|██████    | 2.64G/4.38G [01:55<01:11, 24.3MiB/s]
60%|██████    | 2.64G/4.38G [01:55<01:08, 25.3MiB/s]
60%|██████    | 2.64G/4.38G [01:55<01:10, 24.5MiB/s]
60%|██████    | 2.65G/4.38G [01:56<00:59, 29.1MiB/s]
61%|██████    | 2.65G/4.38G [01:56<01:05, 26.5MiB/s]
61%|██████    | 2.66G/4.38G [01:56<01:01, 27.9MiB/s]
61%|██████    | 2.66G/4.38G [01:56<01:14, 23.2MiB/s]
61%|██████    | 2.67G/4.38G [01:56<00:54, 31.3MiB/s]
61%|██████    | 2.67G/4.38G [01:56<01:00, 28.1MiB/s]
61%|██████    | 2.68G/4.38G [01:57<01:03, 26.7MiB/s]
61%|██████    | 2.68G/4.38G [01:57<00:52, 32.2MiB/s]
61%|██████▏   | 2.69G/4.38G [01:57<00:59, 28.7MiB/s]
61%|██████▏   | 2.69G/4.38G [01:57<01:05, 25.7MiB/s]
61%|██████▏   | 2.69G/4.38G [01:57<01:10, 24.1MiB/s]
62%|██████▏   | 2.70G/4.38G [01:57<01:00, 27.6MiB/s]
62%|██████▏   | 2.70G/4.38G [01:58<01:02, 26.8MiB/s]
62%|██████▏   | 2.70G/4.38G [01:58<01:02, 26.8MiB/s]
62%|██████▏   | 2.71G/4.38G [01:58<01:01, 27.3MiB/s]
62%|██████▏   | 2.72G/4.38G [01:58<00:45, 36.8MiB/s]
62%|██████▏   | 2.72G/4.38G [01:58<00:54, 30.7MiB/s]
62%|██████▏   | 2.72G/4.38G [01:58<01:06, 24.9MiB/s]
62%|██████▏   | 2.73G/4.38G [01:59<01:20, 20.7MiB/s]
62%|██████▏   | 2.73G/4.38G [01:59<01:04, 25.6MiB/s]
63%|██████▎   | 2.74G/4.38G [01:59<00:50, 32.2MiB/s]
63%|██████▎   | 2.75G/4.38G [01:59<00:59, 27.6MiB/s]
63%|██████▎   | 2.75G/4.38G [01:59<00:52, 30.8MiB/s]
63%|██████▎   | 2.75G/4.38G [01:59<01:04, 25.1MiB/s]
63%|██████▎   | 2.76G/4.38G [02:00<00:56, 28.8MiB/s]
63%|██████▎   | 2.76G/4.38G [02:00<01:04, 25.1MiB/s]
63%|██████▎   | 2.76G/4.38G [02:00<01:18, 20.6MiB/s]
63%|██████▎   | 2.77G/4.38G [02:00<01:23, 19.4MiB/s]
63%|██████▎   | 2.77G/4.38G [02:00<01:41, 15.8MiB/s]
63%|██████▎   | 2.78G/4.38G [02:00<01:00, 26.6MiB/s]
63%|██████▎   | 2.78G/4.38G [02:01<00:59, 26.7MiB/s]
64%|██████▎   | 2.78G/4.38G [02:01<00:56, 28.1MiB/s]
64%|██████▎   | 2.79G/4.38G [02:01<01:06, 24.2MiB/s]
64%|██████▎   | 2.79G/4.38G [02:01<00:54, 29.3MiB/s]
64%|██████▍   | 2.79G/4.38G [02:01<01:10, 22.7MiB/s]
64%|██████▍   | 2.80G/4.38G [02:01<00:57, 27.4MiB/s]
64%|██████▍   | 2.81G/4.38G [02:01<00:45, 34.9MiB/s]
64%|██████▍   | 2.81G/4.38G [02:02<00:52, 29.8MiB/s]
64%|██████▍   | 2.82G/4.38G [02:02<01:13, 21.3MiB/s]
64%|██████▍   | 2.82G/4.38G [02:02<01:30, 17.3MiB/s]
65%|██████▍   | 2.83G/4.38G [02:03<01:12, 21.5MiB/s]
65%|██████▍   | 2.83G/4.38G [02:03<01:03, 24.4MiB/s]
65%|██████▍   | 2.84G/4.38G [02:03<01:13, 20.9MiB/s]
65%|██████▍   | 2.84G/4.38G [02:03<01:13, 20.9MiB/s]
65%|██████▍   | 2.84G/4.38G [02:03<01:03, 24.4MiB/s]
65%|██████▍   | 2.85G/4.38G [02:03<01:05, 23.5MiB/s]
65%|██████▌   | 2.85G/4.38G [02:03<00:59, 25.9MiB/s]
65%|██████▌   | 2.85G/4.38G [02:04<00:58, 26.0MiB/s]
65%|██████▌   | 2.86G/4.38G [02:04<01:06, 23.0MiB/s]
65%|██████▌   | 2.86G/4.38G [02:04<00:53, 28.3MiB/s]
65%|██████▌   | 2.87G/4.38G [02:04<00:40, 37.8MiB/s]
66%|██████▌   | 2.87G/4.38G [02:04<00:52, 28.5MiB/s]
66%|██████▌   | 2.88G/4.38G [02:04<00:58, 25.6MiB/s]
66%|██████▌   | 2.88G/4.38G [02:05<01:03, 23.7MiB/s]
66%|██████▌   | 2.88G/4.38G [02:05<00:55, 27.2MiB/s]
66%|██████▌   | 2.89G/4.38G [02:05<00:54, 27.5MiB/s]
66%|██████▌   | 2.89G/4.38G [02:05<00:43, 34.1MiB/s]
66%|██████▌   | 2.90G/4.38G [02:05<00:43, 34.3MiB/s]
66%|██████▋   | 2.90G/4.38G [02:05<00:54, 27.1MiB/s]
66%|██████▋   | 2.91G/4.38G [02:05<00:44, 33.4MiB/s]
67%|██████▋   | 2.91G/4.38G [02:06<00:48, 30.4MiB/s]
67%|██████▋   | 2.92G/4.38G [02:06<00:49, 29.7MiB/s]
67%|██████▋   | 2.92G/4.38G [02:06<01:00, 24.2MiB/s]
67%|██████▋   | 2.93G/4.38G [02:06<01:00, 24.1MiB/s]
67%|██████▋   | 2.93G/4.38G [02:06<01:20, 18.2MiB/s]
67%|██████▋   | 2.93G/4.38G [02:07<01:37, 14.8MiB/s]
67%|██████▋   | 2.94G/4.38G [02:07<01:18, 18.5MiB/s]
67%|██████▋   | 2.94G/4.38G [02:07<00:58, 24.5MiB/s]
67%|██████▋   | 2.95G/4.38G [02:07<01:06, 21.5MiB/s]
67%|██████▋   | 2.95G/4.38G [02:07<01:13, 19.6MiB/s]
67%|██████▋   | 2.95G/4.38G [02:08<01:22, 17.4MiB/s]
67%|██████▋   | 2.95G/4.38G [02:08<01:22, 17.3MiB/s]
68%|██████▊   | 2.96G/4.38G [02:08<01:23, 17.0MiB/s]
68%|██████▊   | 2.96G/4.38G [02:08<01:29, 15.8MiB/s]
68%|██████▊   | 2.97G/4.38G [02:08<01:00, 23.5MiB/s]
68%|██████▊   | 2.97G/4.38G [02:09<01:07, 20.9MiB/s]
68%|██████▊   | 2.98G/4.38G [02:09<00:59, 23.6MiB/s]
68%|██████▊   | 2.98G/4.38G [02:09<01:02, 22.5MiB/s]
68%|██████▊   | 2.99G/4.38G [02:09<00:42, 32.5MiB/s]
68%|██████▊   | 2.99G/4.38G [02:09<00:46, 29.8MiB/s]
68%|██████▊   | 2.99G/4.38G [02:09<00:50, 27.2MiB/s]
68%|██████▊   | 3.00G/4.38G [02:10<00:56, 24.6MiB/s]
69%|██████▊   | 3.00G/4.38G [02:10<00:49, 27.9MiB/s]
69%|██████▊   | 3.00G/4.38G [02:10<00:50, 27.0MiB/s]
69%|██████▊   | 3.01G/4.38G [02:10<00:48, 28.4MiB/s]
69%|██████▉   | 3.02G/4.38G [02:10<00:49, 27.6MiB/s]
69%|██████▉   | 3.02G/4.38G [02:10<00:56, 24.2MiB/s]
69%|██████▉   | 3.03G/4.38G [02:11<00:49, 27.2MiB/s]
69%|██████▉   | 3.03G/4.38G [02:11<00:45, 29.9MiB/s]
69%|██████▉   | 3.04G/4.38G [02:11<00:48, 27.8MiB/s]
69%|██████▉   | 3.04G/4.38G [02:11<00:45, 29.2MiB/s]
70%|██████▉   | 3.05G/4.38G [02:11<00:52, 25.3MiB/s]
70%|██████▉   | 3.05G/4.38G [02:12<01:00, 21.9MiB/s]
70%|██████▉   | 3.05G/4.38G [02:12<01:13, 18.2MiB/s]
70%|██████▉   | 3.06G/4.38G [02:12<00:54, 24.4MiB/s]
70%|███████   | 3.07G/4.38G [02:12<00:45, 28.9MiB/s]
70%|███████   | 3.08G/4.38G [02:12<00:42, 30.4MiB/s]
70%|███████   | 3.08G/4.38G [02:13<00:47, 27.2MiB/s]
70%|███████   | 3.08G/4.38G [02:13<00:50, 25.8MiB/s]
70%|███████   | 3.09G/4.38G [02:13<00:59, 21.7MiB/s]
71%|███████   | 3.09G/4.38G [02:13<00:58, 21.9MiB/s]
71%|███████   | 3.10G/4.38G [02:13<00:59, 21.5MiB/s]
71%|███████   | 3.10G/4.38G [02:14<00:45, 28.4MiB/s]
71%|███████   | 3.11G/4.38G [02:14<00:51, 24.8MiB/s]
71%|███████   | 3.11G/4.38G [02:14<00:43, 29.2MiB/s]
71%|███████   | 3.12G/4.38G [02:14<00:46, 27.4MiB/s]
71%|███████▏  | 3.12G/4.38G [02:14<00:51, 24.6MiB/s]
71%|███████▏  | 3.13G/4.38G [02:15<00:52, 23.7MiB/s]
71%|███████▏  | 3.13G/4.38G [02:15<01:01, 20.4MiB/s]
72%|███████▏  | 3.14G/4.38G [02:15<00:49, 25.2MiB/s]
72%|███████▏  | 3.14G/4.38G [02:15<00:58, 21.1MiB/s]
72%|███████▏  | 3.14G/4.38G [02:15<00:50, 24.6MiB/s]
72%|███████▏  | 3.15G/4.38G [02:16<00:56, 21.9MiB/s]
72%|███████▏  | 3.15G/4.38G [02:16<00:46, 26.3MiB/s]
72%|███████▏  | 3.16G/4.38G [02:16<00:57, 21.5MiB/s]
72%|███████▏  | 3.16G/4.38G [02:16<00:40, 30.3MiB/s]
72%|███████▏  | 3.17G/4.38G [02:16<00:36, 33.1MiB/s]
72%|███████▏  | 3.17G/4.38G [02:16<00:48, 24.9MiB/s]
73%|███████▎  | 3.18G/4.38G [02:17<00:39, 30.3MiB/s]
73%|███████▎  | 3.18G/4.38G [02:17<00:54, 21.9MiB/s]
73%|███████▎  | 3.19G/4.38G [02:17<00:50, 23.5MiB/s]
73%|███████▎  | 3.19G/4.38G [02:17<00:49, 24.1MiB/s]
73%|███████▎  | 3.20G/4.38G [02:17<00:37, 31.7MiB/s]
73%|███████▎  | 3.20G/4.38G [02:18<00:39, 29.6MiB/s]
73%|███████▎  | 3.21G/4.38G [02:18<00:34, 33.7MiB/s]
73%|███████▎  | 3.21G/4.38G [02:18<00:39, 29.3MiB/s]
73%|███████▎  | 3.22G/4.38G [02:18<00:35, 32.5MiB/s]
74%|███████▎  | 3.22G/4.38G [02:18<00:36, 31.4MiB/s]
74%|███████▎  | 3.23G/4.38G [02:18<00:42, 27.3MiB/s]
74%|███████▎  | 3.23G/4.38G [02:18<00:43, 26.4MiB/s]
74%|███████▍  | 3.23G/4.38G [02:19<00:43, 26.6MiB/s]
74%|███████▍  | 3.24G/4.38G [02:19<00:45, 25.3MiB/s]
74%|███████▍  | 3.24G/4.38G [02:19<00:57, 19.8MiB/s]
74%|███████▍  | 3.24G/4.38G [02:19<00:48, 23.7MiB/s]
74%|███████▍  | 3.25G/4.38G [02:19<00:49, 22.7MiB/s]
74%|███████▍  | 3.25G/4.38G [02:19<00:45, 25.0MiB/s]
74%|███████▍  | 3.26G/4.38G [02:20<00:51, 21.9MiB/s]
74%|███████▍  | 3.26G/4.38G [02:20<00:39, 28.5MiB/s]
75%|███████▍  | 3.27G/4.38G [02:20<00:35, 31.3MiB/s]
75%|███████▍  | 3.27G/4.38G [02:20<00:44, 25.1MiB/s]
75%|███████▍  | 3.28G/4.38G [02:20<00:35, 30.9MiB/s]
75%|███████▍  | 3.28G/4.38G [02:20<00:39, 27.6MiB/s]
75%|███████▌  | 3.29G/4.38G [02:21<00:40, 26.9MiB/s]
75%|███████▌  | 3.29G/4.38G [02:21<00:49, 22.2MiB/s]
75%|███████▌  | 3.30G/4.38G [02:21<00:42, 25.8MiB/s]
75%|███████▌  | 3.30G/4.38G [02:21<00:44, 24.1MiB/s]
75%|███████▌  | 3.30G/4.38G [02:21<00:37, 28.6MiB/s]
75%|███████▌  | 3.31G/4.38G [02:21<00:44, 24.0MiB/s]
76%|███████▌  | 3.31G/4.38G [02:22<00:37, 28.9MiB/s]
76%|███████▌  | 3.31G/4.38G [02:22<00:38, 27.4MiB/s]
76%|███████▌  | 3.32G/4.38G [02:22<00:38, 27.7MiB/s]
76%|███████▌  | 3.32G/4.38G [02:22<00:42, 24.6MiB/s]
76%|███████▌  | 3.33G/4.38G [02:22<00:34, 30.4MiB/s]
76%|███████▌  | 3.33G/4.38G [02:22<00:36, 29.1MiB/s]
76%|███████▌  | 3.34G/4.38G [02:23<00:36, 28.7MiB/s]
76%|███████▋  | 3.35G/4.38G [02:23<00:30, 33.9MiB/s]
76%|███████▋  | 3.35G/4.38G [02:23<00:37, 27.6MiB/s]
77%|███████▋  | 3.35G/4.38G [02:23<00:32, 31.8MiB/s]
77%|███████▋  | 3.36G/4.38G [02:23<00:36, 27.8MiB/s]
77%|███████▋  | 3.36G/4.38G [02:23<00:37, 27.2MiB/s]
77%|███████▋  | 3.36G/4.38G [02:24<00:42, 24.0MiB/s]
77%|███████▋  | 3.37G/4.38G [02:24<00:39, 25.8MiB/s]
77%|███████▋  | 3.37G/4.38G [02:24<00:43, 23.1MiB/s]
77%|███████▋  | 3.38G/4.38G [02:24<00:37, 26.5MiB/s]
77%|███████▋  | 3.39G/4.38G [02:24<00:28, 34.7MiB/s]
78%|███████▊  | 3.40G/4.38G [02:24<00:28, 34.4MiB/s]
78%|███████▊  | 3.40G/4.38G [02:25<00:38, 25.8MiB/s]
78%|███████▊  | 3.41G/4.38G [02:25<00:33, 29.3MiB/s]
78%|███████▊  | 3.41G/4.38G [02:25<00:30, 32.3MiB/s]
78%|███████▊  | 3.42G/4.38G [02:25<00:32, 29.4MiB/s]
78%|███████▊  | 3.42G/4.38G [02:25<00:35, 26.7MiB/s]
78%|███████▊  | 3.42G/4.38G [02:26<00:41, 23.2MiB/s]
78%|███████▊  | 3.43G/4.38G [02:26<00:29, 32.4MiB/s]
78%|███████▊  | 3.43G/4.38G [02:26<00:35, 26.8MiB/s]
78%|███████▊  | 3.44G/4.38G [02:26<00:42, 22.1MiB/s]
79%|███████▊  | 3.44G/4.38G [02:26<00:48, 19.3MiB/s]
79%|███████▊  | 3.45G/4.38G [02:26<00:30, 30.6MiB/s]
79%|███████▉  | 3.46G/4.38G [02:27<00:29, 31.9MiB/s]
79%|███████▉  | 3.46G/4.38G [02:27<00:26, 35.2MiB/s]
79%|███████▉  | 3.47G/4.38G [02:27<00:32, 28.5MiB/s]
79%|███████▉  | 3.47G/4.38G [02:27<00:32, 28.2MiB/s]
79%|███████▉  | 3.48G/4.38G [02:28<00:34, 26.5MiB/s]
79%|███████▉  | 3.48G/4.38G [02:28<00:40, 22.3MiB/s]
80%|███████▉  | 3.49G/4.38G [02:28<00:32, 27.2MiB/s]
80%|███████▉  | 3.49G/4.38G [02:28<00:38, 23.2MiB/s]
80%|███████▉  | 3.50G/4.38G [02:28<00:33, 26.5MiB/s]
80%|███████▉  | 3.50G/4.38G [02:28<00:37, 23.7MiB/s]
80%|███████▉  | 3.50G/4.38G [02:29<00:30, 29.1MiB/s]
80%|████████  | 3.51G/4.38G [02:29<00:31, 27.5MiB/s]
80%|████████  | 3.51G/4.38G [02:29<00:28, 30.1MiB/s]
80%|████████  | 3.52G/4.38G [02:29<00:32, 26.6MiB/s]
80%|████████  | 3.52G/4.38G [02:29<00:34, 25.1MiB/s]
81%|████████  | 3.53G/4.38G [02:30<00:29, 28.5MiB/s]
81%|████████  | 3.54G/4.38G [02:30<00:41, 20.1MiB/s]
81%|████████  | 3.54G/4.38G [02:30<00:44, 18.7MiB/s]
81%|████████  | 3.55G/4.38G [02:30<00:34, 24.2MiB/s]
81%|████████  | 3.55G/4.38G [02:31<00:39, 21.1MiB/s]
81%|████████  | 3.56G/4.38G [02:31<00:35, 23.5MiB/s]
81%|████████  | 3.56G/4.38G [02:31<00:38, 21.3MiB/s]
81%|████████▏ | 3.56G/4.38G [02:31<00:28, 28.8MiB/s]
81%|████████▏ | 3.57G/4.38G [02:31<00:29, 27.2MiB/s]
82%|████████▏ | 3.57G/4.38G [02:31<00:26, 30.1MiB/s]
82%|████████▏ | 3.58G/4.38G [02:32<00:28, 27.7MiB/s]
82%|████████▏ | 3.58G/4.38G [02:32<00:31, 25.3MiB/s]
82%|████████▏ | 3.59G/4.38G [02:32<00:27, 29.1MiB/s]
82%|████████▏ | 3.60G/4.38G [02:32<00:31, 24.8MiB/s]
82%|████████▏ | 3.60G/4.38G [02:32<00:32, 23.7MiB/s]
82%|████████▏ | 3.61G/4.38G [02:33<00:31, 25.0MiB/s]
82%|████████▏ | 3.61G/4.38G [02:33<00:35, 21.9MiB/s]
82%|████████▏ | 3.61G/4.38G [02:33<00:27, 27.7MiB/s]
83%|████████▎ | 3.62G/4.38G [02:33<00:33, 23.1MiB/s]
83%|████████▎ | 3.62G/4.38G [02:33<00:25, 29.3MiB/s]
83%|████████▎ | 3.63G/4.38G [02:34<00:30, 25.0MiB/s]
83%|████████▎ | 3.63G/4.38G [02:34<00:36, 20.7MiB/s]
83%|████████▎ | 3.64G/4.38G [02:34<00:26, 27.9MiB/s]
83%|████████▎ | 3.65G/4.38G [02:34<00:21, 33.6MiB/s]
83%|████████▎ | 3.65G/4.38G [02:34<00:25, 29.2MiB/s]
83%|████████▎ | 3.66G/4.38G [02:34<00:23, 31.5MiB/s]
84%|████████▎ | 3.66G/4.38G [02:35<00:27, 26.3MiB/s]
84%|████████▎ | 3.66G/4.38G [02:35<00:28, 25.4MiB/s]
84%|████████▎ | 3.67G/4.38G [02:35<00:31, 22.4MiB/s]
84%|████████▍ | 3.67G/4.38G [02:35<00:30, 23.1MiB/s]
84%|████████▍ | 3.67G/4.38G [02:35<00:32, 21.5MiB/s]
84%|████████▍ | 3.68G/4.38G [02:36<00:38, 18.2MiB/s]
84%|████████▍ | 3.68G/4.38G [02:36<00:44, 15.8MiB/s]
84%|████████▍ | 3.69G/4.38G [02:36<00:33, 20.7MiB/s]
84%|████████▍ | 3.69G/4.38G [02:36<00:38, 18.0MiB/s]
84%|████████▍ | 3.70G/4.38G [02:37<00:28, 23.6MiB/s]
84%|████████▍ | 3.70G/4.38G [02:37<00:32, 20.7MiB/s]
85%|████████▍ | 3.71G/4.38G [02:37<00:35, 19.1MiB/s]
85%|████████▍ | 3.71G/4.38G [02:37<00:36, 18.5MiB/s]
85%|████████▍ | 3.71G/4.38G [02:37<00:24, 27.0MiB/s]
85%|████████▍ | 3.72G/4.38G [02:37<00:25, 25.7MiB/s]
85%|████████▌ | 3.72G/4.38G [02:38<00:20, 31.4MiB/s]
85%|████████▌ | 3.73G/4.38G [02:38<00:20, 32.0MiB/s]
85%|████████▌ | 3.73G/4.38G [02:38<00:21, 29.8MiB/s]
85%|████████▌ | 3.74G/4.38G [02:38<00:20, 31.0MiB/s]
85%|████████▌ | 3.74G/4.38G [02:38<00:21, 29.3MiB/s]
86%|████████▌ | 3.75G/4.38G [02:38<00:18, 34.3MiB/s]
86%|████████▌ | 3.76G/4.38G [02:39<00:18, 33.0MiB/s]
86%|████████▌ | 3.76G/4.38G [02:39<00:21, 28.5MiB/s]
86%|████████▌ | 3.76G/4.38G [02:39<00:19, 30.9MiB/s]
86%|████████▌ | 3.77G/4.38G [02:39<00:21, 28.4MiB/s]
86%|████████▌ | 3.77G/4.38G [02:39<00:19, 31.4MiB/s]
86%|████████▌ | 3.78G/4.38G [02:39<00:26, 22.5MiB/s]
86%|████████▋ | 3.78G/4.38G [02:40<00:23, 25.4MiB/s]
86%|████████▋ | 3.78G/4.38G [02:40<00:26, 22.4MiB/s]
87%|████████▋ | 3.79G/4.38G [02:40<00:23, 25.2MiB/s]
87%|████████▋ | 3.79G/4.38G [02:40<00:25, 22.9MiB/s]
87%|████████▋ | 3.80G/4.38G [02:40<00:24, 24.0MiB/s]
87%|████████▋ | 3.80G/4.38G [02:41<00:34, 16.8MiB/s]
87%|████████▋ | 3.81G/4.38G [02:41<00:28, 20.0MiB/s]
87%|████████▋ | 3.81G/4.38G [02:41<00:32, 17.5MiB/s]
87%|████████▋ | 3.82G/4.38G [02:41<00:22, 25.1MiB/s]
87%|████████▋ | 3.82G/4.38G [02:41<00:22, 24.8MiB/s]
87%|████████▋ | 3.82G/4.38G [02:41<00:20, 27.8MiB/s]
87%|████████▋ | 3.83G/4.38G [02:42<00:22, 25.2MiB/s]
87%|████████▋ | 3.83G/4.38G [02:42<00:21, 26.1MiB/s]
88%|████████▊ | 3.83G/4.38G [02:42<00:23, 22.9MiB/s]
88%|████████▊ | 3.84G/4.38G [02:42<00:21, 25.0MiB/s]
88%|████████▊ | 3.84G/4.38G [02:42<00:21, 25.4MiB/s]
88%|████████▊ | 3.85G/4.38G [02:42<00:18, 29.5MiB/s]
88%|████████▊ | 3.85G/4.38G [02:43<00:23, 23.0MiB/s]
88%|████████▊ | 3.86G/4.38G [02:43<00:21, 24.0MiB/s]
88%|████████▊ | 3.86G/4.38G [02:43<00:22, 23.2MiB/s]
88%|████████▊ | 3.86G/4.38G [02:43<00:22, 22.6MiB/s]
88%|████████▊ | 3.87G/4.38G [02:43<00:20, 24.8MiB/s]
88%|████████▊ | 3.87G/4.38G [02:43<00:24, 21.2MiB/s]
88%|████████▊ | 3.87G/4.38G [02:44<00:34, 14.8MiB/s]
88%|████████▊ | 3.87G/4.38G [02:44<00:33, 15.2MiB/s]
88%|████████▊ | 3.88G/4.38G [02:44<00:29, 17.1MiB/s]
89%|████████▊ | 3.88G/4.38G [02:44<00:19, 25.3MiB/s]
89%|████████▊ | 3.88G/4.38G [02:44<00:19, 24.9MiB/s]
89%|████████▉ | 3.89G/4.38G [02:44<00:18, 26.7MiB/s]
89%|████████▉ | 3.90G/4.38G [02:45<00:15, 30.3MiB/s]
89%|████████▉ | 3.90G/4.38G [02:45<00:19, 25.1MiB/s]
89%|████████▉ | 3.91G/4.38G [02:45<00:19, 24.1MiB/s]
89%|████████▉ | 3.91G/4.38G [02:45<00:23, 20.0MiB/s]
89%|████████▉ | 3.92G/4.38G [02:45<00:20, 22.9MiB/s]
89%|████████▉ | 3.92G/4.38G [02:46<00:22, 20.2MiB/s]
89%|████████▉ | 3.92G/4.38G [02:46<00:32, 14.4MiB/s]
90%|████████▉ | 3.92G/4.38G [02:46<00:35, 12.9MiB/s]
90%|████████▉ | 3.93G/4.38G [02:46<00:29, 15.3MiB/s]
90%|████████▉ | 3.93G/4.38G [02:47<00:28, 15.5MiB/s]
90%|████████▉ | 3.93G/4.38G [02:47<00:35, 12.5MiB/s]
90%|████████▉ | 3.94G/4.38G [02:47<00:22, 19.7MiB/s]
90%|█████████ | 3.94G/4.38G [02:47<00:23, 18.6MiB/s]
90%|█████████ | 3.95G/4.38G [02:48<00:18, 22.7MiB/s]
90%|█████████ | 3.95G/4.38G [02:48<00:22, 19.4MiB/s]
90%|█████████ | 3.96G/4.38G [02:48<00:20, 20.2MiB/s]
90%|█████████ | 3.96G/4.38G [02:48<00:23, 17.7MiB/s]
91%|█████████ | 3.97G/4.38G [02:48<00:15, 26.5MiB/s]
91%|█████████ | 3.97G/4.38G [02:48<00:12, 32.6MiB/s]
91%|█████████ | 3.98G/4.38G [02:49<00:14, 28.5MiB/s]
91%|█████████ | 3.98G/4.38G [02:49<00:13, 28.5MiB/s]
91%|█████████ | 3.99G/4.38G [02:49<00:11, 33.7MiB/s]
91%|█████████ | 3.99G/4.38G [02:50<00:21, 18.3MiB/s]
91%|█████████▏| 4.00G/4.38G [02:50<00:17, 21.7MiB/s]
91%|█████████▏| 4.00G/4.38G [02:50<00:17, 21.0MiB/s]
91%|█████████▏| 4.01G/4.38G [02:50<00:14, 25.5MiB/s]
92%|█████████▏| 4.01G/4.38G [02:50<00:14, 25.8MiB/s]
92%|█████████▏| 4.02G/4.38G [02:50<00:12, 28.2MiB/s]
92%|█████████▏| 4.02G/4.38G [02:51<00:13, 26.8MiB/s]
92%|█████████▏| 4.03G/4.38G [02:51<00:15, 23.1MiB/s]
92%|█████████▏| 4.03G/4.38G [02:51<00:13, 26.4MiB/s]
92%|█████████▏| 4.04G/4.38G [02:51<00:14, 23.6MiB/s]
92%|█████████▏| 4.04G/4.38G [02:51<00:11, 28.9MiB/s]
92%|█████████▏| 4.05G/4.38G [02:51<00:11, 28.4MiB/s]
93%|█████████▎| 4.05G/4.38G [02:52<00:13, 24.3MiB/s]
93%|█████████▎| 4.06G/4.38G [02:52<00:10, 30.5MiB/s]
93%|█████████▎| 4.06G/4.38G [02:52<00:12, 25.2MiB/s]
93%|█████████▎| 4.07G/4.38G [02:52<00:11, 28.6MiB/s]
93%|█████████▎| 4.07G/4.38G [02:52<00:14, 21.7MiB/s]
93%|█████████▎| 4.08G/4.38G [02:53<00:17, 17.6MiB/s]
93%|█████████▎| 4.08G/4.38G [02:53<00:17, 17.7MiB/s]
93%|█████████▎| 4.08G/4.38G [02:53<00:12, 22.9MiB/s]
93%|█████████▎| 4.09G/4.38G [02:53<00:16, 18.4MiB/s]
93%|█████████▎| 4.09G/4.38G [02:54<00:12, 23.3MiB/s]
94%|█████████▎| 4.10G/4.38G [02:54<00:11, 25.5MiB/s]
94%|█████████▎| 4.10G/4.38G [02:54<00:12, 21.9MiB/s]
94%|█████████▍| 4.11G/4.38G [02:54<00:09, 27.7MiB/s]
94%|█████████▍| 4.11G/4.38G [02:54<00:12, 21.7MiB/s]
94%|█████████▍| 4.12G/4.38G [02:54<00:10, 25.2MiB/s]
94%|█████████▍| 4.12G/4.38G [02:55<00:11, 22.0MiB/s]
94%|█████████▍| 4.13G/4.38G [02:55<00:10, 24.9MiB/s]
94%|█████████▍| 4.13G/4.38G [02:55<00:11, 21.4MiB/s]
94%|█████████▍| 4.13G/4.38G [02:55<00:12, 19.8MiB/s]
94%|█████████▍| 4.14G/4.38G [02:56<00:13, 17.9MiB/s]
95%|█████████▍| 4.14G/4.38G [02:56<00:10, 23.1MiB/s]
95%|█████████▍| 4.15G/4.38G [02:56<00:11, 19.8MiB/s]
95%|█████████▍| 4.15G/4.38G [02:56<00:10, 23.0MiB/s]
95%|█████████▍| 4.15G/4.38G [02:56<00:11, 20.4MiB/s]
95%|█████████▍| 4.16G/4.38G [02:56<00:08, 26.9MiB/s]
95%|█████████▌| 4.16G/4.38G [02:57<00:09, 22.7MiB/s]
95%|█████████▌| 4.17G/4.38G [02:57<00:07, 27.7MiB/s]
95%|█████████▌| 4.17G/4.38G [02:57<00:08, 24.3MiB/s]
95%|█████████▌| 4.18G/4.38G [02:57<00:07, 25.7MiB/s]
95%|█████████▌| 4.18G/4.38G [02:57<00:08, 22.5MiB/s]
96%|█████████▌| 4.18G/4.38G [02:57<00:08, 22.6MiB/s]
96%|█████████▌| 4.19G/4.38G [02:58<00:08, 22.0MiB/s]
96%|█████████▌| 4.19G/4.38G [02:58<00:07, 25.5MiB/s]
96%|█████████▌| 4.20G/4.38G [02:58<00:09, 18.8MiB/s]
96%|█████████▌| 4.20G/4.38G [02:58<00:07, 23.2MiB/s]
96%|█████████▌| 4.20G/4.38G [02:58<00:08, 20.5MiB/s]
96%|█████████▌| 4.21G/4.38G [02:59<00:07, 23.5MiB/s]
96%|█████████▌| 4.21G/4.38G [02:59<00:07, 22.6MiB/s]
96%|█████████▋| 4.22G/4.38G [02:59<00:06, 25.8MiB/s]
96%|█████████▋| 4.22G/4.38G [02:59<00:08, 19.0MiB/s]
96%|█████████▋| 4.23G/4.38G [02:59<00:06, 22.8MiB/s]
97%|█████████▋| 4.23G/4.38G [03:00<00:07, 19.2MiB/s]
97%|█████████▋| 4.23G/4.38G [03:00<00:06, 22.3MiB/s]
97%|█████████▋| 4.24G/4.38G [03:00<00:07, 18.4MiB/s]
97%|█████████▋| 4.24G/4.38G [03:00<00:05, 25.2MiB/s]
97%|█████████▋| 4.25G/4.38G [03:00<00:05, 26.1MiB/s]
97%|█████████▋| 4.25G/4.38G [03:00<00:04, 31.4MiB/s]
97%|█████████▋| 4.25G/4.38G [03:01<00:05, 23.9MiB/s]
97%|█████████▋| 4.26G/4.38G [03:01<00:04, 29.2MiB/s]
97%|█████████▋| 4.26G/4.38G [03:01<00:05, 22.2MiB/s]
97%|█████████▋| 4.27G/4.38G [03:01<00:04, 22.9MiB/s]
98%|█████████▊| 4.27G/4.38G [03:01<00:04, 23.5MiB/s]
98%|█████████▊| 4.28G/4.38G [03:01<00:03, 26.7MiB/s]
98%|█████████▊| 4.28G/4.38G [03:02<00:04, 25.0MiB/s]
98%|█████████▊| 4.28G/4.38G [03:02<00:03, 24.2MiB/s]
98%|█████████▊| 4.29G/4.38G [03:02<00:04, 21.4MiB/s]
98%|█████████▊| 4.29G/4.38G [03:02<00:04, 20.3MiB/s]
98%|█████████▊| 4.29G/4.38G [03:02<00:03, 24.6MiB/s]
98%|█████████▊| 4.30G/4.38G [03:02<00:03, 25.6MiB/s]
98%|█████████▊| 4.30G/4.38G [03:03<00:03, 23.6MiB/s]
98%|█████████▊| 4.31G/4.38G [03:03<00:02, 27.5MiB/s]
98%|█████████▊| 4.31G/4.38G [03:03<00:02, 23.8MiB/s]
99%|█████████▊| 4.32G/4.38G [03:03<00:02, 26.9MiB/s]
99%|█████████▉| 4.33G/4.38G [03:03<00:01, 32.8MiB/s]
99%|█████████▉| 4.33G/4.38G [03:03<00:01, 30.0MiB/s]
99%|█████████▉| 4.34G/4.38G [03:04<00:01, 27.1MiB/s]
99%|█████████▉| 4.34G/4.38G [03:04<00:02, 20.3MiB/s]
99%|█████████▉| 4.35G/4.38G [03:04<00:01, 25.6MiB/s]
99%|█████████▉| 4.35G/4.38G [03:04<00:01, 24.8MiB/s]
99%|█████████▉| 4.35G/4.38G [03:04<00:01, 24.7MiB/s]
100%|█████████▉| 4.36G/4.38G [03:05<00:00, 30.4MiB/s]
100%|█████████▉| 4.36G/4.38G [03:05<00:00, 29.4MiB/s]
100%|█████████▉| 4.37G/4.38G [03:05<00:00, 26.6MiB/s]
100%|█████████▉| 4.37G/4.38G [03:05<00:00, 24.1MiB/s]
100%|█████████▉| 4.37G/4.38G [03:06<00:00, 14.8MiB/s]
100%|█████████▉| 4.38G/4.38G [03:06<00:00, 14.4MiB/s]
100%|█████████▉| 4.38G/4.38G [03:06<00:00, 12.5MiB/s]
100%|█████████▉| 4.38G/4.38G [03:06<00:00, 12.4MiB/s]
100%|██████████| 4.38G/4.38G [03:06<00:00, 23.5MiB/s]

Then we will load the csv files.

dataset_path = os.path.join(download_dir, 'flickr30k_processed')
train_data = pd.read_csv(f'{dataset_path}/train.csv', index_col=0)
val_data = pd.read_csv(f'{dataset_path}/val.csv', index_col=0)
test_data = pd.read_csv(f'{dataset_path}/test.csv', index_col=0)
image_col = "image"
text_col = "caption"

We also need to expand the relative image paths to use their absolute local paths.

def path_expander(path, base_folder):
    path_l = path.split(';')
    return ';'.join([os.path.abspath(os.path.join(base_folder, path)) for path in path_l])

train_data[image_col] = train_data[image_col].apply(lambda ele: path_expander(ele, base_folder=dataset_path))
val_data[image_col] = val_data[image_col].apply(lambda ele: path_expander(ele, base_folder=dataset_path))
test_data[image_col] = test_data[image_col].apply(lambda ele: path_expander(ele, base_folder=dataset_path))

Take train_data for example, let’s see how the data look like in the dataframe.

train_data.head()
caption image
0 Two young guys with shaggy hair look at their ... /home/ci/autogluon/docs/tutorials/multimodal/s...
1 Two young White males are outside near many bu... /home/ci/autogluon/docs/tutorials/multimodal/s...
2 Two men in green shirts are standing in a yard /home/ci/autogluon/docs/tutorials/multimodal/s...
3 A man in a blue shirt standing in a garden /home/ci/autogluon/docs/tutorials/multimodal/s...
4 Two friends enjoy time spent together /home/ci/autogluon/docs/tutorials/multimodal/s...

Each row is one image and text pair, implying that they match each other. Since one image corresponds to five captions in the dataset, we copy each image path five times to build the correspondences. We can visualize one image-text pair.

train_data[text_col][0]
'Two young guys with shaggy hair look at their hands while hanging out in the yard'
pil_img = Image(filename=train_data[image_col][0])
display(pil_img)
../../../_images/b012c7e966f6550874ccb85ef9602d483aa89b8623dff9ffcdb0faab8f2ca9ab.jpg

To perform evaluation or semantic search, we need to extract the unique image and text items from text_data and add one label column in the test_data.

test_image_data = pd.DataFrame({image_col: test_data[image_col].unique().tolist()})
test_text_data = pd.DataFrame({text_col: test_data[text_col].unique().tolist()})
test_data_with_label = test_data.copy()
test_label_col = "relevance"
test_data_with_label[test_label_col] = [1] * len(test_data)

Initialize Predictor

To initialize a predictor for image-text matching, we need to set problem_type as image_text_similarity. query and response refer to the two dataframe columns in which two items in one row should match each other. You can set query=text_col and response=image_col, or query=image_col and response=text_col. In image-text matching, query and response are equivalent.

from autogluon.multimodal import MultiModalPredictor
predictor = MultiModalPredictor(
            query=text_col,
            response=image_col,
            problem_type="image_text_similarity",
            eval_metric="recall",
        )
/home/ci/opt/venv/lib/python3.11/site-packages/mmengine/optim/optimizer/zero_optimizer.py:11: DeprecationWarning: `TorchScript` support for functional optimizers is deprecated and will be removed in a future PyTorch release. Consider using the `torch.compile` optimizer instead.
  from torch.distributed.optim import \

By initializing the predictor for image_text_similarity, you have loaded the pretrained CLIP backbone openai/clip-vit-base-patch32.

Directly Evaluate on Test Dataset (Zero-shot)

You may be interested in getting the pretrained model’s performance on your data. Let’s compute the text-to-image and image-to-text retrieval scores.

txt_to_img_scores = predictor.evaluate(
            data=test_data_with_label,
            query_data=test_text_data,
            response_data=test_image_data,
            label=test_label_col,
            cutoffs=[1, 5, 10],
        )
img_to_txt_scores = predictor.evaluate(
            data=test_data_with_label,
            query_data=test_image_data,
            response_data=test_text_data,
            label=test_label_col,
            cutoffs=[1, 5, 10],
        )
print(f"txt_to_img_scores: {txt_to_img_scores}")
print(f"img_to_txt_scores: {img_to_txt_scores}")
txt_to_img_scores: {'recall@1': 0.58964, 'recall@5': 0.83533, 'recall@10': 0.90156}
img_to_txt_scores: {'recall@1': 0.15525, 'recall@5': 0.5708, 'recall@10': 0.7172}

Here we report the recall, which is the eval_metric in initializing the predictor above. One cutoff value means using the top k retrieved items to calculate the score. You may find that the text-to-image recalls are much higher than the image-to-text recalls. This is because each image is paired with five texts. In image-to-text retrieval, the upper bound of recall@1 is 20%, which means that the top-1 text is correct, but there are totally five texts to retrieve.

Finetune Predictor

After measuring the pretrained performance, we can finetune the model on our dataset to see whether we can get improvements. For a quick demo, here we set the time limit to 180 seconds.

predictor.fit(
            train_data=train_data,
            tuning_data=val_data,
            time_limit=180,
        )
No path specified. Models will be saved in: "AutogluonModels/ag-20241108_031234"
=================== System Info ===================
AutoGluon Version:  1.1.2b20241108
Python Version:     3.11.9
Operating System:   Linux
Platform Machine:   x86_64
Platform Version:   #1 SMP Tue Sep 24 10:00:37 UTC 2024
CPU Count:          8
Pytorch Version:    2.4.1+cu121
CUDA Version:       12.1
Memory Avail:       26.82 GB / 30.95 GB (86.7%)
Disk Space Avail:   172.58 GB / 255.99 GB (67.4%)
===================================================
AutoMM starts to create your model. ✨✨✨

To track the learning progress, you can open a terminal and launch Tensorboard:
    ```shell
    # Assume you have installed tensorboard
    tensorboard --logdir /home/ci/autogluon/docs/tutorials/multimodal/semantic_matching/AutogluonModels/ag-20241108_031234
    ```
INFO: Seed set to 0
GPU Count: 1
GPU Count to be Used: 1
GPU 0 Name: Tesla T4
GPU 0 Memory: 0.56GB/15.0GB (Used/Total)
INFO: Using 16bit Automatic Mixed Precision (AMP)
INFO: GPU available: True (cuda), used: True
INFO: TPU available: False, using: 0 TPU cores
INFO: HPU available: False, using: 0 HPUs
INFO: LOCAL_RANK: 0 - CUDA_VISIBLE_DEVICES: [0]
INFO: 
  | Name              | Type                      | Params | Mode 
------------------------------------------------------------------------
0 | query_model       | CLIPForImageText          | 151 M  | train
1 | response_model    | CLIPForImageText          | 151 M  | train
2 | validation_metric | CustomHitRate             | 0      | train
3 | loss_func         | MultiNegativesSoftmaxLoss | 0      | train
------------------------------------------------------------------------
151 M     Trainable params
0         Non-trainable params
151 M     Total params
605.109   Total estimated model params size (MB)
INFO: Time limit reached. Elapsed time is 0:03:00. Signaling Trainer to stop.
INFO: Epoch 0, global step 389: 'val_recall' reached 0.56302 (best 0.56302), saving model to '/home/ci/autogluon/docs/tutorials/multimodal/semantic_matching/AutogluonModels/ag-20241108_031234/epoch=0-step=389.ckpt' as top 3
Start to fuse 1 checkpoints via the greedy soup algorithm.
AutoMM has created your model. 🎉🎉🎉

To load the model, use the code below:
    ```python
    from autogluon.multimodal import MultiModalPredictor
    predictor = MultiModalPredictor.load("/home/ci/autogluon/docs/tutorials/multimodal/semantic_matching/AutogluonModels/ag-20241108_031234")
    ```

If you are not satisfied with the model, try to increase the training time, 
adjust the hyperparameters (https://auto.gluon.ai/stable/tutorials/multimodal/advanced_topics/customization.html),
or post issues on GitHub (https://github.com/autogluon/autogluon/issues).
<autogluon.multimodal.predictor.MultiModalPredictor at 0x7fbdfc7fc390>

Evaluate the Finetuned Model on the Test Dataset

Now Let’s evaluate the finetuned model. Similarly, we also compute the recalls of text-to-image and image-to-text retrievals.

txt_to_img_scores = predictor.evaluate(
            data=test_data_with_label,
            query_data=test_text_data,
            response_data=test_image_data,
            label=test_label_col,
            cutoffs=[1, 5, 10],
        )
img_to_txt_scores = predictor.evaluate(
            data=test_data_with_label,
            query_data=test_image_data,
            response_data=test_text_data,
            label=test_label_col,
            cutoffs=[1, 5, 10],
        )
print(f"txt_to_img_scores: {txt_to_img_scores}")
print(f"img_to_txt_scores: {img_to_txt_scores}")
txt_to_img_scores: {'recall@1': 0.70088, 'recall@5': 0.91076, 'recall@10': 0.95278}
img_to_txt_scores: {'recall@1': 0.16905, 'recall@5': 0.673, 'recall@10': 0.8194}

We can observe large improvements over the zero-shot predictor. This means that finetuning CLIP on our customized data may help achieve better performance.

Predict Whether Image and Text Match

Whether finetuned or not, the predictor can predict whether image and text pairs match.

pred = predictor.predict(test_data.head(5))
print(pred)
0    1
1    1
2    1
3    1
4    1
dtype: int64

Predict Matching Probabilities

The predictor can also return to you the matching probabilities.

proba = predictor.predict_proba(test_data.head(5))
print(proba)
0         1
0  0.341900  0.658100
1  0.326389  0.673611
2  0.346046  0.653954
3  0.343145  0.656855
4  0.328434  0.671566

The second column is the probability of being a match.

Extract Embeddings

Another common user case is to extract image and text embeddings.

image_embeddings = predictor.extract_embedding({image_col: test_image_data[image_col][:5].tolist()})
print(image_embeddings.shape) 
(5, 512)
text_embeddings = predictor.extract_embedding({text_col: test_text_data[text_col][:5].tolist()})
print(text_embeddings.shape)
(5, 512)

Other Examples

You may go to AutoMM Examples to explore other examples about AutoMM.

Customization

To learn how to customize AutoMM, please refer to Customize AutoMM.