Merge remote-tracking branch 'origin/main'

rfgzuid · Jun 12, 2024 · 81ace80 · 81ace80
2 parents 851cce3 + 006b648
commit 81ace80
Show file tree

Hide file tree

Showing 243 changed files with 229 additions and 110 deletions.
diff --git a/Test Data/2371_crop/1000.png b/Test Data/2371_crop/1000.png
diff --git a/Test Data/2371_crop/1001.png b/Test Data/2371_crop/1001.png
diff --git a/Test Data/2371_crop/1002.png b/Test Data/2371_crop/1002.png
diff --git a/Test Data/2371_crop/1003.png b/Test Data/2371_crop/1003.png
diff --git a/Test Data/2371_crop/1004.png b/Test Data/2371_crop/1004.png
diff --git a/Test Data/2371_crop/1005.png b/Test Data/2371_crop/1005.png
diff --git a/Test Data/2371_crop/1006.png b/Test Data/2371_crop/1006.png
diff --git a/Test Data/2371_crop/1007.png b/Test Data/2371_crop/1007.png
diff --git a/Test Data/2371_crop/1008.png b/Test Data/2371_crop/1008.png
diff --git a/Test Data/2371_crop/1009.png b/Test Data/2371_crop/1009.png
diff --git a/Test Data/2371_crop/1010.png b/Test Data/2371_crop/1010.png
diff --git a/Test Data/2371_crop/1011.png b/Test Data/2371_crop/1011.png
diff --git a/Test Data/2371_crop/1012.png b/Test Data/2371_crop/1012.png
diff --git a/Test Data/2371_crop/1013.png b/Test Data/2371_crop/1013.png
diff --git a/Test Data/2371_crop/1014.png b/Test Data/2371_crop/1014.png
diff --git a/Test Data/2371_crop/1015.png b/Test Data/2371_crop/1015.png
diff --git a/Test Data/2371_crop/1016.png b/Test Data/2371_crop/1016.png
diff --git a/Test Data/2371_crop/1236.png b/Test Data/2371_crop/1236.png
diff --git a/Test Data/2371_crop/1237.png b/Test Data/2371_crop/1237.png
diff --git a/Test Data/2371_crop/1238.png b/Test Data/2371_crop/1238.png
diff --git a/Test Data/2371_crop/1239.png b/Test Data/2371_crop/1239.png
diff --git a/Test Data/2371_crop/1240.png b/Test Data/2371_crop/1240.png
diff --git a/Test Data/2371_crop/1241.png b/Test Data/2371_crop/1241.png
diff --git a/Test Data/2371_crop/1242.png b/Test Data/2371_crop/1242.png
diff --git a/Test Data/2371_crop/1243.png b/Test Data/2371_crop/1243.png
diff --git a/Test Data/2371_crop/1244.png b/Test Data/2371_crop/1244.png
diff --git a/Test Data/2371_crop/1245.png b/Test Data/2371_crop/1245.png
diff --git a/Test Data/2371_crop/1246.png b/Test Data/2371_crop/1246.png
diff --git a/Test Data/2371_crop/1247.png b/Test Data/2371_crop/1247.png
diff --git a/Test Data/2371_crop/1248.png b/Test Data/2371_crop/1248.png
diff --git a/Test Data/2371_crop/1249.png b/Test Data/2371_crop/1249.png
diff --git a/Test Data/2371_crop/1250.png b/Test Data/2371_crop/1250.png
diff --git a/Test Data/2371_crop/1251.png b/Test Data/2371_crop/1251.png
diff --git a/Test Data/2371_crop/1252.png b/Test Data/2371_crop/1252.png
diff --git a/Test Data/2371_crop/1253.png b/Test Data/2371_crop/1253.png
diff --git a/Test Data/2371_crop/1254.png b/Test Data/2371_crop/1254.png
diff --git a/Test Data/2371_crop/1255.png b/Test Data/2371_crop/1255.png
diff --git a/Test Data/2371_crop/1256.png b/Test Data/2371_crop/1256.png
diff --git a/Test Data/2371_crop/145100/516.png b/Test Data/2371_crop/145100/516.png
diff --git a/Test Data/2371_crop/145100/517.png b/Test Data/2371_crop/145100/517.png
diff --git a/Test Data/2371_crop/145100/518.png b/Test Data/2371_crop/145100/518.png
diff --git a/Test Data/2371_crop/145100/520.png b/Test Data/2371_crop/145100/520.png
diff --git a/Test Data/2371_crop/145100/521.png b/Test Data/2371_crop/145100/521.png
diff --git a/Test Data/2371_crop/145100/522.png b/Test Data/2371_crop/145100/522.png
diff --git a/Test Data/2371_crop/145100/523.png b/Test Data/2371_crop/145100/523.png
diff --git a/Test Data/2371_crop/145100/524.png b/Test Data/2371_crop/145100/524.png
diff --git a/Test Data/2371_crop/145100/525.png b/Test Data/2371_crop/145100/525.png
diff --git a/Test Data/2371_crop/145100/527.png b/Test Data/2371_crop/145100/527.png
diff --git a/Test Data/2371_crop/145100/528.png b/Test Data/2371_crop/145100/528.png
diff --git a/Test Data/2371_crop/145100/529.png b/Test Data/2371_crop/145100/529.png
diff --git a/Test Data/2371_crop/145100/530.png b/Test Data/2371_crop/145100/530.png
diff --git a/Test Data/2371_crop/145100/531.png b/Test Data/2371_crop/145100/531.png
diff --git a/Test Data/2371_crop/145100/532.png b/Test Data/2371_crop/145100/532.png
diff --git a/Test Data/2371_crop/145100/533.png b/Test Data/2371_crop/145100/533.png
diff --git a/Test Data/2371_crop/145100/534.png b/Test Data/2371_crop/145100/534.png
diff --git a/Test Data/2371_crop/145100/536.png b/Test Data/2371_crop/145100/536.png
diff --git a/Test Data/2371_crop/1476.png b/Test Data/2371_crop/1476.png
diff --git a/Test Data/2371_crop/1477.png b/Test Data/2371_crop/1477.png
diff --git a/Test Data/2371_crop/1478.png b/Test Data/2371_crop/1478.png
diff --git a/Test Data/2371_crop/1479.png b/Test Data/2371_crop/1479.png
diff --git a/Test Data/2371_crop/1480.png b/Test Data/2371_crop/1480.png
diff --git a/Test Data/2371_crop/1481.png b/Test Data/2371_crop/1481.png
diff --git a/Test Data/2371_crop/1482.png b/Test Data/2371_crop/1482.png
diff --git a/Test Data/2371_crop/1483.png b/Test Data/2371_crop/1483.png
diff --git a/Test Data/2371_crop/1484.png b/Test Data/2371_crop/1484.png
diff --git a/Test Data/2371_crop/1485.png b/Test Data/2371_crop/1485.png
diff --git a/Test Data/2371_crop/1486.png b/Test Data/2371_crop/1486.png
diff --git a/Test Data/2371_crop/1487.png b/Test Data/2371_crop/1487.png
diff --git a/Test Data/2371_crop/1488.png b/Test Data/2371_crop/1488.png
diff --git a/Test Data/2371_crop/1489.png b/Test Data/2371_crop/1489.png
diff --git a/Test Data/2371_crop/1490.png b/Test Data/2371_crop/1490.png
diff --git a/Test Data/2371_crop/1491.png b/Test Data/2371_crop/1491.png
diff --git a/Test Data/2371_crop/1492.png b/Test Data/2371_crop/1492.png
diff --git a/Test Data/2371_crop/1493.png b/Test Data/2371_crop/1493.png
diff --git a/Test Data/2371_crop/1494.png b/Test Data/2371_crop/1494.png
diff --git a/Test Data/2371_crop/1495.png b/Test Data/2371_crop/1495.png
diff --git a/Test Data/2371_crop/1496.png b/Test Data/2371_crop/1496.png
diff --git a/Test Data/2371_crop/516.png b/Test Data/2371_crop/516.png
diff --git a/Test Data/2371_crop/517.png b/Test Data/2371_crop/517.png
diff --git a/Test Data/2371_crop/518.png b/Test Data/2371_crop/518.png
diff --git a/Test Data/2371_crop/519.png b/Test Data/2371_crop/519.png
diff --git a/Test Data/2371_crop/520.png b/Test Data/2371_crop/520.png
diff --git a/Test Data/2371_crop/521.png b/Test Data/2371_crop/521.png
diff --git a/Test Data/2371_crop/522.png b/Test Data/2371_crop/522.png
diff --git a/Test Data/2371_crop/523.png b/Test Data/2371_crop/523.png
diff --git a/Test Data/2371_crop/524.png b/Test Data/2371_crop/524.png
diff --git a/Test Data/2371_crop/525.png b/Test Data/2371_crop/525.png
diff --git a/Test Data/2371_crop/526.png b/Test Data/2371_crop/526.png
diff --git a/Test Data/2371_crop/527.png b/Test Data/2371_crop/527.png
diff --git a/Test Data/2371_crop/528.png b/Test Data/2371_crop/528.png
diff --git a/Test Data/2371_crop/529.png b/Test Data/2371_crop/529.png
diff --git a/Test Data/2371_crop/530.png b/Test Data/2371_crop/530.png
diff --git a/Test Data/2371_crop/531.png b/Test Data/2371_crop/531.png
diff --git a/Test Data/2371_crop/532.png b/Test Data/2371_crop/532.png
diff --git a/Test Data/2371_crop/533.png b/Test Data/2371_crop/533.png
diff --git a/Test Data/2371_crop/534.png b/Test Data/2371_crop/534.png
diff --git a/Test Data/2371_crop/535.png b/Test Data/2371_crop/535.png
diff --git a/Test Data/2371_crop/536.png b/Test Data/2371_crop/536.png
diff --git a/Test Data/2371_crop/756.png b/Test Data/2371_crop/756.png
diff --git a/Test Data/2371_crop/757.png b/Test Data/2371_crop/757.png
diff --git a/Test Data/2371_crop/758.png b/Test Data/2371_crop/758.png
diff --git a/Test Data/2371_crop/759.png b/Test Data/2371_crop/759.png
diff --git a/Test Data/2371_crop/760.png b/Test Data/2371_crop/760.png
diff --git a/Test Data/2371_crop/761.png b/Test Data/2371_crop/761.png
diff --git a/Test Data/2371_crop/762.png b/Test Data/2371_crop/762.png
diff --git a/Test Data/2371_crop/763.png b/Test Data/2371_crop/763.png
diff --git a/Test Data/2371_crop/764.png b/Test Data/2371_crop/764.png
diff --git a/Test Data/2371_crop/765.png b/Test Data/2371_crop/765.png
diff --git a/Test Data/2371_crop/766.png b/Test Data/2371_crop/766.png
diff --git a/Test Data/2371_crop/767.png b/Test Data/2371_crop/767.png
diff --git a/Test Data/2371_crop/768.png b/Test Data/2371_crop/768.png
diff --git a/Test Data/2371_crop/769.png b/Test Data/2371_crop/769.png
diff --git a/Test Data/2371_crop/770.png b/Test Data/2371_crop/770.png
diff --git a/Test Data/2371_crop/771.png b/Test Data/2371_crop/771.png
diff --git a/Test Data/2371_crop/772.png b/Test Data/2371_crop/772.png
diff --git a/Test Data/2371_crop/773.png b/Test Data/2371_crop/773.png
diff --git a/Test Data/2371_crop/774.png b/Test Data/2371_crop/774.png
diff --git a/Test Data/2371_crop/775.png b/Test Data/2371_crop/775.png
diff --git a/Test Data/2371_crop/776.png b/Test Data/2371_crop/776.png
diff --git a/Test Data/2371_crop/996.png b/Test Data/2371_crop/996.png
diff --git a/Test Data/2371_crop/997.png b/Test Data/2371_crop/997.png
diff --git a/Test Data/2371_crop/998.png b/Test Data/2371_crop/998.png
diff --git a/Test Data/2371_crop/999.png b/Test Data/2371_crop/999.png
diff --git a/Test Data/2372_crop/145157/540.png b/Test Data/2372_crop/145157/540.png
diff --git a/Test Data/2372_crop/145157/541.png b/Test Data/2372_crop/145157/541.png
diff --git a/Test Data/2372_crop/145157/542.png b/Test Data/2372_crop/145157/542.png
diff --git a/Test Data/2372_crop/145157/543.png b/Test Data/2372_crop/145157/543.png
diff --git a/Test Data/2372_crop/145157/544.png b/Test Data/2372_crop/145157/544.png
diff --git a/Test Data/2372_crop/145157/545.png b/Test Data/2372_crop/145157/545.png
diff --git a/Test Data/2372_crop/145157/546.png b/Test Data/2372_crop/145157/546.png
diff --git a/Test Data/2372_crop/145157/548.png b/Test Data/2372_crop/145157/548.png
diff --git a/Test Data/2372_crop/145157/549.png b/Test Data/2372_crop/145157/549.png
diff --git a/Test Data/2372_crop/145157/550.png b/Test Data/2372_crop/145157/550.png
diff --git a/Test Data/2372_crop/145157/551.png b/Test Data/2372_crop/145157/551.png
diff --git a/Test Data/2372_crop/145157/552.png b/Test Data/2372_crop/145157/552.png
diff --git a/Test Data/2372_crop/145157/553.png b/Test Data/2372_crop/145157/553.png
diff --git a/Test Data/2372_crop/145157/555.png b/Test Data/2372_crop/145157/555.png
diff --git a/Test Data/2372_crop/145157/556.png b/Test Data/2372_crop/145157/556.png
diff --git a/Test Data/2372_crop/145157/557.png b/Test Data/2372_crop/145157/557.png
diff --git a/Test Data/2372_crop/145157/558.png b/Test Data/2372_crop/145157/558.png
diff --git a/Test Data/2372_crop/145157/559.png b/Test Data/2372_crop/145157/559.png
diff --git a/Test Data/2372_crop/145157/560.png b/Test Data/2372_crop/145157/560.png
diff --git a/Test Data/2372_crop/145158/780.png b/Test Data/2372_crop/145158/780.png
diff --git a/Test Data/2372_crop/145158/781.png b/Test Data/2372_crop/145158/781.png
diff --git a/Test Data/2372_crop/145158/782.png b/Test Data/2372_crop/145158/782.png
diff --git a/Test Data/2372_crop/145158/783.png b/Test Data/2372_crop/145158/783.png
diff --git a/Test Data/2372_crop/145158/784.png b/Test Data/2372_crop/145158/784.png
diff --git a/Test Data/2372_crop/145158/785.png b/Test Data/2372_crop/145158/785.png
diff --git a/Test Data/2372_crop/145158/786.png b/Test Data/2372_crop/145158/786.png
diff --git a/Test Data/2372_crop/145158/787.png b/Test Data/2372_crop/145158/787.png
diff --git a/Test Data/2372_crop/145158/788.png b/Test Data/2372_crop/145158/788.png
diff --git a/Test Data/2372_crop/145158/789.png b/Test Data/2372_crop/145158/789.png
diff --git a/Test Data/2372_crop/145158/790.png b/Test Data/2372_crop/145158/790.png
diff --git a/Test Data/2372_crop/145158/791.png b/Test Data/2372_crop/145158/791.png
diff --git a/Test Data/2372_crop/145158/792.png b/Test Data/2372_crop/145158/792.png
diff --git a/Test Data/2372_crop/145158/793.png b/Test Data/2372_crop/145158/793.png
diff --git a/Test Data/2372_crop/145158/794.png b/Test Data/2372_crop/145158/794.png
diff --git a/Test Data/2372_crop/145158/795.png b/Test Data/2372_crop/145158/795.png
diff --git a/Test Data/2372_crop/145158/796.png b/Test Data/2372_crop/145158/796.png
diff --git a/Test Data/2372_crop/145158/797.png b/Test Data/2372_crop/145158/797.png
diff --git a/Test Data/2372_crop/145158/798.png b/Test Data/2372_crop/145158/798.png
diff --git a/Test Data/2372_crop/145158/799.png b/Test Data/2372_crop/145158/799.png
diff --git a/Test Data/2372_crop/145158/800.png b/Test Data/2372_crop/145158/800.png
diff --git a/Test Data/2372_crop/145159/1020.png b/Test Data/2372_crop/145159/1020.png
diff --git a/Test Data/2372_crop/145159/1021.png b/Test Data/2372_crop/145159/1021.png
diff --git a/Test Data/2372_crop/145159/1022.png b/Test Data/2372_crop/145159/1022.png
diff --git a/Test Data/2372_crop/145159/1023.png b/Test Data/2372_crop/145159/1023.png
diff --git a/Test Data/2372_crop/145159/1024.png b/Test Data/2372_crop/145159/1024.png
diff --git a/Test Data/2372_crop/145159/1025.png b/Test Data/2372_crop/145159/1025.png
diff --git a/Test Data/2372_crop/145159/1026.png b/Test Data/2372_crop/145159/1026.png
diff --git a/Test Data/2372_crop/145159/1027.png b/Test Data/2372_crop/145159/1027.png
diff --git a/Test Data/2372_crop/145159/1028.png b/Test Data/2372_crop/145159/1028.png
diff --git a/Test Data/2372_crop/145159/1029.png b/Test Data/2372_crop/145159/1029.png
diff --git a/Test Data/2372_crop/145159/1030.png b/Test Data/2372_crop/145159/1030.png
diff --git a/Test Data/2372_crop/145159/1031.png b/Test Data/2372_crop/145159/1031.png
diff --git a/Test Data/2372_crop/145159/1032.png b/Test Data/2372_crop/145159/1032.png
diff --git a/Test Data/2372_crop/145159/1033.png b/Test Data/2372_crop/145159/1033.png
diff --git a/Test Data/2372_crop/145159/1034.png b/Test Data/2372_crop/145159/1034.png
diff --git a/Test Data/2372_crop/145159/1035.png b/Test Data/2372_crop/145159/1035.png
diff --git a/Test Data/2372_crop/145159/1036.png b/Test Data/2372_crop/145159/1036.png
diff --git a/Test Data/2372_crop/145159/1037.png b/Test Data/2372_crop/145159/1037.png
diff --git a/Test Data/2372_crop/145159/1038.png b/Test Data/2372_crop/145159/1038.png
diff --git a/Test Data/2372_crop/145159/1039.png b/Test Data/2372_crop/145159/1039.png
diff --git a/Test Data/2372_crop/145159/1040.png b/Test Data/2372_crop/145159/1040.png
diff --git a/Test Data/2372_crop/145200/1260.png b/Test Data/2372_crop/145200/1260.png
diff --git a/Test Data/2372_crop/145200/1261.png b/Test Data/2372_crop/145200/1261.png
diff --git a/Test Data/2372_crop/145200/1262.png b/Test Data/2372_crop/145200/1262.png
diff --git a/Test Data/2372_crop/145200/1263.png b/Test Data/2372_crop/145200/1263.png
diff --git a/Test Data/2372_crop/145200/1264.png b/Test Data/2372_crop/145200/1264.png
diff --git a/Test Data/2372_crop/145200/1265.png b/Test Data/2372_crop/145200/1265.png
diff --git a/Test Data/2372_crop/145200/1266.png b/Test Data/2372_crop/145200/1266.png
diff --git a/Test Data/2372_crop/145200/1267.png b/Test Data/2372_crop/145200/1267.png
diff --git a/Test Data/2372_crop/145200/1268.png b/Test Data/2372_crop/145200/1268.png
diff --git a/Test Data/2372_crop/145200/1269.png b/Test Data/2372_crop/145200/1269.png
diff --git a/Test Data/2372_crop/145200/1270.png b/Test Data/2372_crop/145200/1270.png
diff --git a/Test Data/2372_crop/145200/1271.png b/Test Data/2372_crop/145200/1271.png
diff --git a/Test Data/2372_crop/145200/1272.png b/Test Data/2372_crop/145200/1272.png
diff --git a/Test Data/2372_crop/145200/1273.png b/Test Data/2372_crop/145200/1273.png
diff --git a/Test Data/2372_crop/145200/1274.png b/Test Data/2372_crop/145200/1274.png
diff --git a/Test Data/2372_crop/145200/1275.png b/Test Data/2372_crop/145200/1275.png
diff --git a/Test Data/2372_crop/145200/1276.png b/Test Data/2372_crop/145200/1276.png
diff --git a/Test Data/2372_crop/145200/1277.png b/Test Data/2372_crop/145200/1277.png
diff --git a/Test Data/2372_crop/145200/1278.png b/Test Data/2372_crop/145200/1278.png
diff --git a/Test Data/2372_crop/145200/1279.png b/Test Data/2372_crop/145200/1279.png
diff --git a/Test Data/2372_crop/145200/1280.png b/Test Data/2372_crop/145200/1280.png
diff --git a/Test Data/crazy/10297.png b/Test Data/crazy/10297.png
diff --git a/Test Data/crazy/10298.png b/Test Data/crazy/10298.png
diff --git a/Test Data/crazy/10337.png b/Test Data/crazy/10337.png
diff --git a/Test Data/crazy/10338.png b/Test Data/crazy/10338.png
diff --git a/Test Data/crazy/10377.png b/Test Data/crazy/10377.png
diff --git a/Test Data/crazy/10378.png b/Test Data/crazy/10378.png
diff --git a/Test Data/crazy/10417.png b/Test Data/crazy/10417.png
diff --git a/Test Data/crazy/10418.png b/Test Data/crazy/10418.png
diff --git a/Test Data/crazy/10457.png b/Test Data/crazy/10457.png
diff --git a/Test Data/crazy/10458.png b/Test Data/crazy/10458.png
diff --git a/Test Data/eersteseconde/516.png b/Test Data/eersteseconde/516.png
diff --git a/Test Data/eersteseconde/517.png b/Test Data/eersteseconde/517.png
diff --git a/Test Data/eersteseconde/518.png b/Test Data/eersteseconde/518.png
diff --git a/Test Data/eersteseconde/520.png b/Test Data/eersteseconde/520.png
diff --git a/Test Data/eersteseconde/521.png b/Test Data/eersteseconde/521.png
diff --git a/Test Data/eersteseconde/522.png b/Test Data/eersteseconde/522.png
diff --git a/Test Data/eersteseconde/523.png b/Test Data/eersteseconde/523.png
diff --git a/Test Data/eersteseconde/524.png b/Test Data/eersteseconde/524.png
diff --git a/Test Data/eersteseconde/525.png b/Test Data/eersteseconde/525.png
diff --git a/Test Data/eersteseconde/527.png b/Test Data/eersteseconde/527.png
diff --git a/Test Data/eersteseconde/528.png b/Test Data/eersteseconde/528.png
diff --git a/Test Data/eersteseconde/529.png b/Test Data/eersteseconde/529.png
diff --git a/Test Data/eersteseconde/530.png b/Test Data/eersteseconde/530.png
diff --git a/Test Data/eersteseconde/531.png b/Test Data/eersteseconde/531.png
diff --git a/Test Data/eersteseconde/532.png b/Test Data/eersteseconde/532.png
diff --git a/Test Data/eersteseconde/533.png b/Test Data/eersteseconde/533.png
diff --git a/Test Data/eersteseconde/534.png b/Test Data/eersteseconde/534.png
diff --git a/Test Data/eersteseconde/536.png b/Test Data/eersteseconde/536.png
diff --git a/Test Data/ff/225.png b/Test Data/ff/225.png
diff --git a/Test Data/ff/226.png b/Test Data/ff/226.png
diff --git a/Test Data/gg/vortex_pair_particles_1.tif b/Test Data/gg/vortex_pair_particles_1.tif
diff --git a/Test Data/gg/vortex_pair_particles_2.tif b/Test Data/gg/vortex_pair_particles_2.tif
diff --git a/Test Data/hh/00101.jpg b/Test Data/hh/00101.jpg
diff --git a/Test Data/hh/00102.jpg b/Test Data/hh/00102.jpg
diff --git a/experiments/PIV Test.ipynb b/experiments/PIV Test.ipynb
diff --git a/experiments/hybrid.gif b/experiments/hybrid.gif
diff --git a/src/SIV_library/advanced.py b/src/SIV_library/advanced.py
@@ -62,6 +62,7 @@ def ctf_optical(optical: OpticalFlow, num_passes: int = 3, scale_factor: float =
         resize.apply_to = ['a', 'b']  # apply the resize transform to both images in the pair
         warp = Warp(x, y, u, v)
 
+        optical.dataset.img_shape = size
         optical.dataset.transforms = [resize, warp]
 
         _, _, du, dv = optical.run()

diff --git a/src/SIV_library/lib.py b/src/SIV_library/lib.py
@@ -11,15 +11,18 @@
 
 class SIVDataset(Dataset):
     def __init__(self, folder: str, transforms: list | None = None, device: str = "cpu") -> None:
-        # assume the files are sorted and all have the correct file type
+        # assume the files have the correct file type
         filenames = [os.path.join(folder, name) for name in os.listdir(folder)]
         filenames.sort(key=lambda x: int(os.path.split(x)[-1].split('.')[0]))
+
         self.img_pairs = list(zip(filenames[:-1], filenames[1:]))
+        self.idx = [int(os.path.split(x)[-1].split('.')[0]) for x in filenames]
 
-        self.img_shape = cv2.imread(filenames[0], cv2.IMREAD_GRAYSCALE).shape
         self.transforms = transforms
         self.device = device
 
+        self.img_shape = cv2.imread(filenames[0], cv2.IMREAD_GRAYSCALE).shape
+
     def __len__(self) -> int:
         return len(self.img_pairs)