Table 2. Benchmark results of the ESPINNet.

Image size (pixels)	Batch size	Time/batch (ms)	GPU memory (GB)
2048 × 2048	1	220	2.4
2048 × 2048	4	378	8.1
2048 × 2048	8	589	15.9
2048 × 2048	16	1358	31.2
1024 × 1024	1	195	0.8
1024 × 1024	4	252	2.2
1024 × 1024	8	278	4.1
1024 × 1024	16	380	7.9
512 × 512	4	208	0.7
512 × 512	8	239	1.2
512 × 512	16	250	2.1
512 × 512	32	277	4.0