diff --git a/.gitattributes b/.gitattributes index a6344aac8c09253b3b630fb776ae94478aa0275b..c178812293740955a63cd89e90a3b506e7f6aaa5 100644 --- a/.gitattributes +++ b/.gitattributes @@ -33,3 +33,38 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text *.zip filter=lfs diff=lfs merge=lfs -text *.zst filter=lfs diff=lfs merge=lfs -text *tfevents* filter=lfs diff=lfs merge=lfs -text +dataset/gallery/american_flag01.jpg filter=lfs diff=lfs merge=lfs -text +dataset/gallery/american_flag03.jpg filter=lfs diff=lfs merge=lfs -text +dataset/gallery/american_flag05.jpg filter=lfs diff=lfs merge=lfs -text +dataset/gallery/american_flag07.jpg filter=lfs diff=lfs merge=lfs -text +dataset/gallery/american_flag08.jpg filter=lfs diff=lfs merge=lfs -text +dataset/gallery/american_flag13.jpg filter=lfs diff=lfs merge=lfs -text +dataset/gallery/american_flag14.jpg filter=lfs diff=lfs merge=lfs -text +dataset/gallery/british_flag06.jpg filter=lfs diff=lfs merge=lfs -text +dataset/gallery/british_flag11.jpg filter=lfs diff=lfs merge=lfs -text +dataset/gallery/british_flag12.jpg filter=lfs diff=lfs merge=lfs -text +dataset/gallery/british_flag13.jpg filter=lfs diff=lfs merge=lfs -text +dataset/gallery/british_flag14.jpg filter=lfs diff=lfs merge=lfs -text +dataset/gallery/british_flag17.jpg filter=lfs diff=lfs merge=lfs -text +dataset/gallery/greek_flag01.jpg filter=lfs diff=lfs merge=lfs -text +dataset/gallery/greek_flag04.jpg filter=lfs diff=lfs merge=lfs -text +dataset/gallery/greek_flag14.jpg filter=lfs diff=lfs merge=lfs -text +dataset/gallery/tattoo_Screenshot[[:space:]]from[[:space:]]2025-02-07[[:space:]]14-12-57.png filter=lfs diff=lfs merge=lfs -text +dataset/gallery/tattoo_Screenshot[[:space:]]from[[:space:]]2025-02-07[[:space:]]14-13-10.png filter=lfs diff=lfs merge=lfs -text +dataset/gallery/tattoo_Screenshot[[:space:]]from[[:space:]]2025-02-07[[:space:]]14-13-31.png filter=lfs diff=lfs merge=lfs -text +dataset/gallery/tattoo_Screenshot[[:space:]]from[[:space:]]2025-02-07[[:space:]]14-13-34.png filter=lfs diff=lfs merge=lfs -text +dataset/gallery/tattoo_Screenshot[[:space:]]from[[:space:]]2025-02-07[[:space:]]14-13-47.png filter=lfs diff=lfs merge=lfs -text +dataset/gallery/tattoo_Screenshot[[:space:]]from[[:space:]]2025-02-07[[:space:]]14-15-19.png filter=lfs diff=lfs merge=lfs -text +dataset/gallery/tattoo_Screenshot[[:space:]]from[[:space:]]2025-02-07[[:space:]]14-17-34.png filter=lfs diff=lfs merge=lfs -text +dataset/gallery/tattoo_Screenshot[[:space:]]from[[:space:]]2025-02-07[[:space:]]14-18-37.png filter=lfs diff=lfs merge=lfs -text +dataset/gallery/tattoo_Screenshot[[:space:]]from[[:space:]]2025-02-07[[:space:]]14-20-16.png filter=lfs diff=lfs merge=lfs -text +dataset/gallery/tattoo_Screenshot[[:space:]]from[[:space:]]2025-02-07[[:space:]]14-20-18.png filter=lfs diff=lfs merge=lfs -text +dataset/gallery/tattoo_Screenshot[[:space:]]from[[:space:]]2025-02-07[[:space:]]14-20-35.png filter=lfs diff=lfs merge=lfs -text +dataset/gallery/tattoo_Screenshot[[:space:]]from[[:space:]]2025-02-07[[:space:]]14-21-24.png filter=lfs diff=lfs merge=lfs -text +dataset/gallery/tattoo_Screenshot[[:space:]]from[[:space:]]2025-02-07[[:space:]]14-21-37.png filter=lfs diff=lfs merge=lfs -text +dataset/gallery/tattoo_Screenshot[[:space:]]from[[:space:]]2025-02-07[[:space:]]14-50-22.png filter=lfs diff=lfs merge=lfs -text +dataset/gallery/tattoo_Screenshot[[:space:]]from[[:space:]]2025-02-07[[:space:]]14-50-29.png filter=lfs diff=lfs merge=lfs -text +dataset/gallery/tattoo_Screenshot[[:space:]]from[[:space:]]2025-02-07[[:space:]]14-50-49.png filter=lfs diff=lfs merge=lfs -text +dataset/gallery/tattoo_Screenshot[[:space:]]from[[:space:]]2025-02-07[[:space:]]14-51-23.png filter=lfs diff=lfs merge=lfs -text +dataset/gallery/tattoo_Screenshot[[:space:]]from[[:space:]]2025-02-07[[:space:]]14-53-30.png filter=lfs diff=lfs merge=lfs -text +dataset/gallery/tattoo_Screenshot[[:space:]]from[[:space:]]2025-02-07[[:space:]]14-53-43.png filter=lfs diff=lfs merge=lfs -text diff --git a/dataset/gallery/american_flag01.jpg b/dataset/gallery/american_flag01.jpg new file mode 100644 index 0000000000000000000000000000000000000000..6e90c4975ad79c48414d8f8019d923486800978f --- /dev/null +++ b/dataset/gallery/american_flag01.jpg @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b08441e937bb8ca4ec9c606b67566fe11fc7395071650b503a4e93b72b584522 +size 107337 diff --git a/dataset/gallery/american_flag02.jpg b/dataset/gallery/american_flag02.jpg new file mode 100644 index 0000000000000000000000000000000000000000..f6a5bf5205d049ea47d90c9ee4b295eb63a915d2 Binary files /dev/null and b/dataset/gallery/american_flag02.jpg differ diff --git a/dataset/gallery/american_flag03.jpg b/dataset/gallery/american_flag03.jpg new file mode 100644 index 0000000000000000000000000000000000000000..60704d7217730fb5695dd72fdb3dd48da4c6f69d --- /dev/null +++ b/dataset/gallery/american_flag03.jpg @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:2622bad4da0e52845139260fb252d78d0a29b9ee3d4be5af4c341be7ecf6b8ff +size 756043 diff --git a/dataset/gallery/american_flag04.jpg b/dataset/gallery/american_flag04.jpg new file mode 100644 index 0000000000000000000000000000000000000000..fbc6e266b84f666248d12d8ed8359224bc0d9ad4 Binary files /dev/null and b/dataset/gallery/american_flag04.jpg differ diff --git a/dataset/gallery/american_flag05.jpg b/dataset/gallery/american_flag05.jpg new file mode 100644 index 0000000000000000000000000000000000000000..8d9025c7c62c70b762a203ed33f97e3dcdc0e447 --- /dev/null +++ b/dataset/gallery/american_flag05.jpg @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b4fc74170b7040afd177b64630435c93ae583fdd7180f726f733f62d3c278317 +size 281335 diff --git a/dataset/gallery/american_flag06.jpg b/dataset/gallery/american_flag06.jpg new file mode 100644 index 0000000000000000000000000000000000000000..0f13a8eec159427f2bb60766d33a3dd28087ef3d Binary files /dev/null and b/dataset/gallery/american_flag06.jpg differ diff --git a/dataset/gallery/american_flag07.jpg b/dataset/gallery/american_flag07.jpg new file mode 100644 index 0000000000000000000000000000000000000000..56f9c4d8700f51d57edc10d5389a290077b07ca5 --- /dev/null +++ b/dataset/gallery/american_flag07.jpg @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6afd40d0ebe8abc65c684e688bcbe0665f9ef3d1c2c6c20a17e0cc2b856e2e44 +size 112055 diff --git a/dataset/gallery/american_flag08.jpg b/dataset/gallery/american_flag08.jpg new file mode 100644 index 0000000000000000000000000000000000000000..04487f76ed0c8ab2dc566c5c62b954427cd700ba --- /dev/null +++ b/dataset/gallery/american_flag08.jpg @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c677854b28f37b1f49062366bd8e4c8390ad951454603a7c4958493b2e827697 +size 548586 diff --git a/dataset/gallery/american_flag09.jpg b/dataset/gallery/american_flag09.jpg new file mode 100644 index 0000000000000000000000000000000000000000..b2ccee14909078a5f18393bf1c49b9c4c5dd7c3f Binary files /dev/null and b/dataset/gallery/american_flag09.jpg differ diff --git a/dataset/gallery/american_flag10.jpg b/dataset/gallery/american_flag10.jpg new file mode 100644 index 0000000000000000000000000000000000000000..f60e4e6f97982ca52f1c9ba7fb1cde2f3d162ac3 Binary files /dev/null and b/dataset/gallery/american_flag10.jpg differ diff --git a/dataset/gallery/american_flag11.jpg b/dataset/gallery/american_flag11.jpg new file mode 100644 index 0000000000000000000000000000000000000000..5a99d60cc9beff484c1c25bca14e7debedeb08f2 Binary files /dev/null and b/dataset/gallery/american_flag11.jpg differ diff --git a/dataset/gallery/american_flag12.jpg b/dataset/gallery/american_flag12.jpg new file mode 100644 index 0000000000000000000000000000000000000000..f3740936f2b5db52caa8735fedcdcecbf2d67a60 Binary files /dev/null and b/dataset/gallery/american_flag12.jpg differ diff --git a/dataset/gallery/american_flag13.jpg b/dataset/gallery/american_flag13.jpg new file mode 100644 index 0000000000000000000000000000000000000000..b098450b402465412cb60f2ff8deceda240977e4 --- /dev/null +++ b/dataset/gallery/american_flag13.jpg @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e3c08009d1fdd77c34efdec5d4e5ac2468bea39a09e926e286403cadf932d472 +size 236517 diff --git a/dataset/gallery/american_flag14.jpg b/dataset/gallery/american_flag14.jpg new file mode 100644 index 0000000000000000000000000000000000000000..0265f9f9d7da32281434821606db9cb739d90ff8 --- /dev/null +++ b/dataset/gallery/american_flag14.jpg @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:f8f03881cab6432e74e6d75718d1d75ca9a3bbc606f8f2a4631e47b6dbbdb6c2 +size 480471 diff --git a/dataset/gallery/american_flag15.jpg b/dataset/gallery/american_flag15.jpg new file mode 100644 index 0000000000000000000000000000000000000000..56b01ddc4a8bfb4639688a10e500043b8ed7be2b Binary files /dev/null and b/dataset/gallery/american_flag15.jpg differ diff --git a/dataset/gallery/british_flag01.jpg b/dataset/gallery/british_flag01.jpg new file mode 100644 index 0000000000000000000000000000000000000000..89fce942aafafb52a37eff89e88f1494335a6c23 Binary files /dev/null and b/dataset/gallery/british_flag01.jpg differ diff --git a/dataset/gallery/british_flag04.jpg b/dataset/gallery/british_flag04.jpg new file mode 100644 index 0000000000000000000000000000000000000000..1a1c41c6e8735d19449f9a0804eef460d9a8dddf Binary files /dev/null and b/dataset/gallery/british_flag04.jpg differ diff --git a/dataset/gallery/british_flag05.jpg b/dataset/gallery/british_flag05.jpg new file mode 100644 index 0000000000000000000000000000000000000000..659b2c8ff44469bd46410af93c9de5216dd037dd Binary files /dev/null and b/dataset/gallery/british_flag05.jpg differ diff --git a/dataset/gallery/british_flag06.jpg b/dataset/gallery/british_flag06.jpg new file mode 100644 index 0000000000000000000000000000000000000000..4c7e963d00dda868213316f20cd051cf1291d390 --- /dev/null +++ b/dataset/gallery/british_flag06.jpg @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:058f73266c2638736cb331b7f66db5224f77a4b79489ac6554cf1e9fddff54bc +size 121085 diff --git a/dataset/gallery/british_flag08.jpg b/dataset/gallery/british_flag08.jpg new file mode 100644 index 0000000000000000000000000000000000000000..8ffc92acec00a0b1a34a3825fc8c46e245241e41 Binary files /dev/null and b/dataset/gallery/british_flag08.jpg differ diff --git a/dataset/gallery/british_flag09.jpg b/dataset/gallery/british_flag09.jpg new file mode 100644 index 0000000000000000000000000000000000000000..1548546a8a3844a537e26f2d23634ccd7ebeb90d Binary files /dev/null and b/dataset/gallery/british_flag09.jpg differ diff --git a/dataset/gallery/british_flag10.jpg b/dataset/gallery/british_flag10.jpg new file mode 100644 index 0000000000000000000000000000000000000000..441338a337f27355f0d31ac6a204dccdbc44e2e0 Binary files /dev/null and b/dataset/gallery/british_flag10.jpg differ diff --git a/dataset/gallery/british_flag11.jpg b/dataset/gallery/british_flag11.jpg new file mode 100644 index 0000000000000000000000000000000000000000..cc0806c25c594eb9f6ac8cbd4e1722ff921ff453 --- /dev/null +++ b/dataset/gallery/british_flag11.jpg @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:5ebcb4e1ad4ba58c5c5bf57020cdc7a926fa7b0ccc0aecd4fa544cd35c9bc54a +size 147957 diff --git a/dataset/gallery/british_flag12.jpg b/dataset/gallery/british_flag12.jpg new file mode 100644 index 0000000000000000000000000000000000000000..fb02deeaab9f6ba255087a54aeb0b96c2f6814b3 --- /dev/null +++ b/dataset/gallery/british_flag12.jpg @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7b7ad9ef3d5bf716b8aaf0b9d314ce3edf39abf7d88187cf74a122a3d4d94081 +size 873671 diff --git a/dataset/gallery/british_flag13.jpg b/dataset/gallery/british_flag13.jpg new file mode 100644 index 0000000000000000000000000000000000000000..8c3c882c2a17d3fa97d14fa63c3278a6e24a01a8 --- /dev/null +++ b/dataset/gallery/british_flag13.jpg @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7a105f0a8b7681d01ec0768336339ff68cce8ae750791614d2cc8d16173789ed +size 272142 diff --git a/dataset/gallery/british_flag14.jpg b/dataset/gallery/british_flag14.jpg new file mode 100644 index 0000000000000000000000000000000000000000..cbd35a73a4e5bbe5476d67a0ee922a5ec7adb22e --- /dev/null +++ b/dataset/gallery/british_flag14.jpg @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:2e3ddb87029e1f5c722c71fc3fec267b259f75eae9409820bb5b9e182f0f3afb +size 609514 diff --git a/dataset/gallery/british_flag15.jpg b/dataset/gallery/british_flag15.jpg new file mode 100644 index 0000000000000000000000000000000000000000..edb3b1823a4acc10632db8337ce3beb6e9bca4f2 Binary files /dev/null and b/dataset/gallery/british_flag15.jpg differ diff --git a/dataset/gallery/british_flag16.jpg b/dataset/gallery/british_flag16.jpg new file mode 100644 index 0000000000000000000000000000000000000000..d10ad0695418171bf76e84df56b3353bf3a2a2b4 Binary files /dev/null and b/dataset/gallery/british_flag16.jpg differ diff --git a/dataset/gallery/british_flag17.jpg b/dataset/gallery/british_flag17.jpg new file mode 100644 index 0000000000000000000000000000000000000000..2820b4d07154a2a8e3da8caf710549842079efbf --- /dev/null +++ b/dataset/gallery/british_flag17.jpg @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e8662177e8a5e5e2bd281959f9bb8ca9227e2fbc71e64e2435c3b701f7b1be1f +size 325611 diff --git a/dataset/gallery/british_flag18.jpg b/dataset/gallery/british_flag18.jpg new file mode 100644 index 0000000000000000000000000000000000000000..7ceed346346c715a6be3f332568406a3b96649de Binary files /dev/null and b/dataset/gallery/british_flag18.jpg differ diff --git a/dataset/gallery/british_flag19.jpg b/dataset/gallery/british_flag19.jpg new file mode 100644 index 0000000000000000000000000000000000000000..46f28837c722e50d3639fd02169129b477a4185f Binary files /dev/null and b/dataset/gallery/british_flag19.jpg differ diff --git a/dataset/gallery/british_flag20.jpg b/dataset/gallery/british_flag20.jpg new file mode 100644 index 0000000000000000000000000000000000000000..855ffc32671fb32e4bd02bcdb2d2e1f9d7023e6c Binary files /dev/null and b/dataset/gallery/british_flag20.jpg differ diff --git a/dataset/gallery/greek_flag01.jpg b/dataset/gallery/greek_flag01.jpg new file mode 100644 index 0000000000000000000000000000000000000000..bd5dd84db0499e9b5da416ae65fafe927cfadbbe --- /dev/null +++ b/dataset/gallery/greek_flag01.jpg @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:3545224971600281465d5bfb5c68e02a85d427e472f886f54f3c92ee501959da +size 282690 diff --git a/dataset/gallery/greek_flag02.jpg b/dataset/gallery/greek_flag02.jpg new file mode 100644 index 0000000000000000000000000000000000000000..ab7b27141ce20ef6db519ada5bd8dde9043fb663 Binary files /dev/null and b/dataset/gallery/greek_flag02.jpg differ diff --git a/dataset/gallery/greek_flag03.jpg b/dataset/gallery/greek_flag03.jpg new file mode 100644 index 0000000000000000000000000000000000000000..fd072716931abdf78156ecd37c3f7afaa20a4c9e Binary files /dev/null and b/dataset/gallery/greek_flag03.jpg differ diff --git a/dataset/gallery/greek_flag04.jpg b/dataset/gallery/greek_flag04.jpg new file mode 100644 index 0000000000000000000000000000000000000000..bd5dd84db0499e9b5da416ae65fafe927cfadbbe --- /dev/null +++ b/dataset/gallery/greek_flag04.jpg @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:3545224971600281465d5bfb5c68e02a85d427e472f886f54f3c92ee501959da +size 282690 diff --git a/dataset/gallery/greek_flag05.jpg b/dataset/gallery/greek_flag05.jpg new file mode 100644 index 0000000000000000000000000000000000000000..520f4bdce2fb14ff34a2d5b9522d9ecf9f6b391d Binary files /dev/null and b/dataset/gallery/greek_flag05.jpg differ diff --git a/dataset/gallery/greek_flag06.jpg b/dataset/gallery/greek_flag06.jpg new file mode 100644 index 0000000000000000000000000000000000000000..520f4bdce2fb14ff34a2d5b9522d9ecf9f6b391d Binary files /dev/null and b/dataset/gallery/greek_flag06.jpg differ diff --git a/dataset/gallery/greek_flag07.jpg b/dataset/gallery/greek_flag07.jpg new file mode 100644 index 0000000000000000000000000000000000000000..1fe27ad074c730eb3bfb75c5beeb1c9b9e56b952 Binary files /dev/null and b/dataset/gallery/greek_flag07.jpg differ diff --git a/dataset/gallery/greek_flag08.jpg b/dataset/gallery/greek_flag08.jpg new file mode 100644 index 0000000000000000000000000000000000000000..1fe27ad074c730eb3bfb75c5beeb1c9b9e56b952 Binary files /dev/null and b/dataset/gallery/greek_flag08.jpg differ diff --git a/dataset/gallery/greek_flag09.jpg b/dataset/gallery/greek_flag09.jpg new file mode 100644 index 0000000000000000000000000000000000000000..e70bab901577db39a617858d114e8bcf5c6252b5 Binary files /dev/null and b/dataset/gallery/greek_flag09.jpg differ diff --git a/dataset/gallery/greek_flag10.jpg b/dataset/gallery/greek_flag10.jpg new file mode 100644 index 0000000000000000000000000000000000000000..e70bab901577db39a617858d114e8bcf5c6252b5 Binary files /dev/null and b/dataset/gallery/greek_flag10.jpg differ diff --git a/dataset/gallery/greek_flag11.jpg b/dataset/gallery/greek_flag11.jpg new file mode 100644 index 0000000000000000000000000000000000000000..2c23df57f631d968eb9cdd0ecceedfa1cb7ed1b1 Binary files /dev/null and b/dataset/gallery/greek_flag11.jpg differ diff --git a/dataset/gallery/greek_flag12.jpg b/dataset/gallery/greek_flag12.jpg new file mode 100644 index 0000000000000000000000000000000000000000..e78ba8c783207c71a74c95e8e49c45d1fe021197 Binary files /dev/null and b/dataset/gallery/greek_flag12.jpg differ diff --git a/dataset/gallery/greek_flag13.jpg b/dataset/gallery/greek_flag13.jpg new file mode 100644 index 0000000000000000000000000000000000000000..b209325d8aa925761f01b155e656ba446328762d Binary files /dev/null and b/dataset/gallery/greek_flag13.jpg differ diff --git a/dataset/gallery/greek_flag14.jpg b/dataset/gallery/greek_flag14.jpg new file mode 100644 index 0000000000000000000000000000000000000000..e37c7dbc62481190efd6d90e53d09f49409b0de6 --- /dev/null +++ b/dataset/gallery/greek_flag14.jpg @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a003515825538f194de607a10ff9f05518229f44a9b72c59acd3dee9568d6688 +size 2723386 diff --git a/dataset/gallery/greek_flag15.jpg b/dataset/gallery/greek_flag15.jpg new file mode 100644 index 0000000000000000000000000000000000000000..2923aa88e89427e6c62f6470b5625bd18924e641 Binary files /dev/null and b/dataset/gallery/greek_flag15.jpg differ diff --git a/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-12-53.png b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-12-53.png new file mode 100644 index 0000000000000000000000000000000000000000..9b43f3410bf2bd888ac57f10b52f0f7b6cef15f6 Binary files /dev/null and b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-12-53.png differ diff --git a/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-12-57.png b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-12-57.png new file mode 100644 index 0000000000000000000000000000000000000000..2b089063ef69b27d33f1d9b2368974d71163ca10 --- /dev/null +++ b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-12-57.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:00ae8b0b9718214ab97378324811384f8ca16978e0f02421728e74a3c5802e90 +size 110074 diff --git a/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-13-00.png b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-13-00.png new file mode 100644 index 0000000000000000000000000000000000000000..2c4ea10d82875e7b000f7f344d5dd23d430ec3f5 Binary files /dev/null and b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-13-00.png differ diff --git a/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-13-03.png b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-13-03.png new file mode 100644 index 0000000000000000000000000000000000000000..7d6e5e5fb35c42fc5983b972d6c6a3ac05887a3a Binary files /dev/null and b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-13-03.png differ diff --git a/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-13-07.png b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-13-07.png new file mode 100644 index 0000000000000000000000000000000000000000..e1b70c802a5eeceb5e79b5e6860863ec8dd14e70 Binary files /dev/null and b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-13-07.png differ diff --git a/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-13-10.png b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-13-10.png new file mode 100644 index 0000000000000000000000000000000000000000..1ffb543a261fc3d8b5aff0fe05d57d12f16e327e --- /dev/null +++ b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-13-10.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:4b1ab719bf33aa6c095e3a09c5c23ce78272b4d623895b63b8e99d467b288cfa +size 959439 diff --git a/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-13-14.png b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-13-14.png new file mode 100644 index 0000000000000000000000000000000000000000..aec77c77b262d4a7d60bccd6caea7a1b6df3abf0 Binary files /dev/null and b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-13-14.png differ diff --git a/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-13-18.png b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-13-18.png new file mode 100644 index 0000000000000000000000000000000000000000..42f09bc0f0fb4b26107cb20e2591d0c0f149de86 Binary files /dev/null and b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-13-18.png differ diff --git a/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-13-21.png b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-13-21.png new file mode 100644 index 0000000000000000000000000000000000000000..b303f13e6b3d722b24824a7b15b96a8eb2d5bc51 Binary files /dev/null and b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-13-21.png differ diff --git a/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-13-24.png b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-13-24.png new file mode 100644 index 0000000000000000000000000000000000000000..0e349bc49fed2093c5fc34ad502a568c33a7360a Binary files /dev/null and b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-13-24.png differ diff --git a/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-13-31.png b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-13-31.png new file mode 100644 index 0000000000000000000000000000000000000000..59a5c19993595a8daad3491a5b15a2584c69cfeb --- /dev/null +++ b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-13-31.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b60189700859eba08ae34f9135a9f6bdd0118e862b76fb6d7256051624e127d3 +size 207097 diff --git a/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-13-34.png b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-13-34.png new file mode 100644 index 0000000000000000000000000000000000000000..f1bd5143a96b74294aec44edb75b125dfaf1a49f --- /dev/null +++ b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-13-34.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:bc7ea38e2b42e6a3c757bf5f6e63570d91d5e21cdea4ef09b61327366dfead80 +size 101259 diff --git a/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-13-36.png b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-13-36.png new file mode 100644 index 0000000000000000000000000000000000000000..7ddbf32104846090a94457707bda3062295dd866 Binary files /dev/null and b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-13-36.png differ diff --git a/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-13-47.png b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-13-47.png new file mode 100644 index 0000000000000000000000000000000000000000..d5ba4dee18d63fd409ed770a1259570080061612 --- /dev/null +++ b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-13-47.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:f7792f66a868ab37a2f257de51e946401cbdd778267a575b00ecae5e8871d836 +size 121495 diff --git a/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-13-49.png b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-13-49.png new file mode 100644 index 0000000000000000000000000000000000000000..f22e7cc9d6585a9dad6921ef4d695d9504143374 Binary files /dev/null and b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-13-49.png differ diff --git a/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-15-19.png b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-15-19.png new file mode 100644 index 0000000000000000000000000000000000000000..c6f1bec82ff42f857e8be236564e36c7e01ccfb4 --- /dev/null +++ b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-15-19.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:38d1420b27172c43336779b495505c632eb391fe61c22c49171b2224589d6737 +size 178162 diff --git a/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-16-28.png b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-16-28.png new file mode 100644 index 0000000000000000000000000000000000000000..7126638f14a2452c4e157ee744af15df879ddc51 Binary files /dev/null and b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-16-28.png differ diff --git a/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-16-56.png b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-16-56.png new file mode 100644 index 0000000000000000000000000000000000000000..03ef578b64149d0e8d96629799282d5090750a00 Binary files /dev/null and b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-16-56.png differ diff --git a/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-16-59.png b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-16-59.png new file mode 100644 index 0000000000000000000000000000000000000000..91d62653816582745802837c9ed9b5828e6114e7 Binary files /dev/null and b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-16-59.png differ diff --git a/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-17-24.png b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-17-24.png new file mode 100644 index 0000000000000000000000000000000000000000..4f65cb5293de0543bc373951554e29239e733dab Binary files /dev/null and b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-17-24.png differ diff --git a/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-17-34.png b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-17-34.png new file mode 100644 index 0000000000000000000000000000000000000000..bdbb8cad50809d502810aa6e2f6e6a61bd952adb --- /dev/null +++ b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-17-34.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:f4d6cdd3ff3c97a5e67a4a57528e98254bbc00acaacfd13cc04a94e3781cfc21 +size 970201 diff --git a/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-18-37.png b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-18-37.png new file mode 100644 index 0000000000000000000000000000000000000000..ebebc4c9e21061ddf6bdf945bb39a184794e8834 --- /dev/null +++ b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-18-37.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a10b408e6691aed63e16ed538a5938b74894b26da4de5d82497309f6dfb36bc7 +size 167054 diff --git a/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-20-16.png b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-20-16.png new file mode 100644 index 0000000000000000000000000000000000000000..de31ff35353c44dd83e4bca05137bec724aedaa0 --- /dev/null +++ b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-20-16.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:3155fe86c5bc4fb0c2d1dffdbd5ececad263eba4e537376242e5449eaae77fab +size 172999 diff --git a/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-20-18.png b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-20-18.png new file mode 100644 index 0000000000000000000000000000000000000000..1a5348f8c2f059efed4c03e5a69c3094c77abd15 --- /dev/null +++ b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-20-18.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:3f73bd89c99da9014846e3e6fc6a4f96acf07f019d4c17eb1f876c373f348200 +size 561492 diff --git a/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-20-23.png b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-20-23.png new file mode 100644 index 0000000000000000000000000000000000000000..0d47be2159aad406d4c4984849d886b407a8e3f5 Binary files /dev/null and b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-20-23.png differ diff --git a/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-20-25.png b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-20-25.png new file mode 100644 index 0000000000000000000000000000000000000000..04c6e01ab5d478c50720e510110f023220681d37 Binary files /dev/null and b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-20-25.png differ diff --git a/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-20-35.png b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-20-35.png new file mode 100644 index 0000000000000000000000000000000000000000..a8974c1355aea8e704ea512e67613794d7be4b32 --- /dev/null +++ b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-20-35.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1f6b640fc2e634808530c8355e1b5fee3aaee4a2fef48e60f62b18e564876657 +size 125985 diff --git a/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-21-24.png b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-21-24.png new file mode 100644 index 0000000000000000000000000000000000000000..2e1839920d6fccd24dacbf00eb696ccf0d54997c --- /dev/null +++ b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-21-24.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:4abb909e819a7d185612b52b56f60f1a2c37f6d934a4552eebe29daf2dde2b56 +size 165718 diff --git a/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-21-27.png b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-21-27.png new file mode 100644 index 0000000000000000000000000000000000000000..61e6eecab51c6d4f57f1ad2e585365c0aceb038b Binary files /dev/null and b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-21-27.png differ diff --git a/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-21-37.png b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-21-37.png new file mode 100644 index 0000000000000000000000000000000000000000..0220c0d044b9c34a9c66c901d38d471cd0ac49b3 --- /dev/null +++ b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-21-37.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:89995b4ef0a84b11cd39f1d4485c0586759dcf2648d0339f25e17e958ac4ed66 +size 560077 diff --git a/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-49-25.png b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-49-25.png new file mode 100644 index 0000000000000000000000000000000000000000..8d4c1436e6517e54151a7fba8ffb9d5a8a97ac4b Binary files /dev/null and b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-49-25.png differ diff --git a/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-49-46.png b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-49-46.png new file mode 100644 index 0000000000000000000000000000000000000000..7b7d74813dca77fc9a8536ef5d38a519e791b151 Binary files /dev/null and b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-49-46.png differ diff --git a/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-50-16.png b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-50-16.png new file mode 100644 index 0000000000000000000000000000000000000000..da80ab4b543f937cd38e907c1a57a4e3f88521b0 Binary files /dev/null and b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-50-16.png differ diff --git a/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-50-18.png b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-50-18.png new file mode 100644 index 0000000000000000000000000000000000000000..3111593f04c24d2ea6e9ef4aca37445534435658 Binary files /dev/null and b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-50-18.png differ diff --git a/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-50-22.png b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-50-22.png new file mode 100644 index 0000000000000000000000000000000000000000..a26983801f5d52adf7660b4a87d7eb841149b47d --- /dev/null +++ b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-50-22.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:095d80285f350742d2320d84d2b7a6440be4aba7495ce1eefe07a7a41c74eb8f +size 143637 diff --git a/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-50-29.png b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-50-29.png new file mode 100644 index 0000000000000000000000000000000000000000..91dc75491f7a472ab58c32cb0e53bb022bfee035 --- /dev/null +++ b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-50-29.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6352c6346d686d0e2de744f7f6d1071937be4535279b1a8529dfc059f4db9958 +size 179114 diff --git a/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-50-49.png b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-50-49.png new file mode 100644 index 0000000000000000000000000000000000000000..9a1b9c150af0b6ac7068a6a271fce0f40286176a --- /dev/null +++ b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-50-49.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:8d85f59cf68019bd8a6583910feb61899538db84123a4df132de0a1c1b4ac988 +size 135358 diff --git a/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-50-57.png b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-50-57.png new file mode 100644 index 0000000000000000000000000000000000000000..2d4e46ed0dc51bbdd47f0943f2e348ff9a4559ab Binary files /dev/null and b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-50-57.png differ diff --git a/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-51-15.png b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-51-15.png new file mode 100644 index 0000000000000000000000000000000000000000..a703eb9ceee6cb179ccd45f132f2691b9d3ccdf1 Binary files /dev/null and b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-51-15.png differ diff --git a/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-51-23.png b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-51-23.png new file mode 100644 index 0000000000000000000000000000000000000000..dcbe0ab29322f96d6e5db80041469331a567b52b --- /dev/null +++ b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-51-23.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1f9d97803f0ed12069e19dd1d9017e5e5732484553ec0515d74b197e6357fe22 +size 102666 diff --git a/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-51-38.png b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-51-38.png new file mode 100644 index 0000000000000000000000000000000000000000..04d1aedf9662ea75bb3d34823eaa8b9c822fc8e2 Binary files /dev/null and b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-51-38.png differ diff --git a/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-52-41.png b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-52-41.png new file mode 100644 index 0000000000000000000000000000000000000000..1fb63a3612b093d0ca1dbfbeb093d5c602a32235 Binary files /dev/null and b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-52-41.png differ diff --git a/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-52-46.png b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-52-46.png new file mode 100644 index 0000000000000000000000000000000000000000..17a9f7dc5f0a8b63899f02e25059f478a159bf8f Binary files /dev/null and b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-52-46.png differ diff --git a/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-53-12.png b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-53-12.png new file mode 100644 index 0000000000000000000000000000000000000000..0379b12cd7c5874b51db1a61920aeeb5f9ac6cf0 Binary files /dev/null and b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-53-12.png differ diff --git a/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-53-30.png b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-53-30.png new file mode 100644 index 0000000000000000000000000000000000000000..ae799ba7d2d79f7b560244fddebeec70680873e9 --- /dev/null +++ b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-53-30.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:4e55d340f63192f4dfa827ab3d80d2b79e6e81a7987b0a05da47a0931db8d1bc +size 126954 diff --git a/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-53-43.png b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-53-43.png new file mode 100644 index 0000000000000000000000000000000000000000..9ac81bc24bb46d54322b11d1239170e4936b2c66 --- /dev/null +++ b/dataset/gallery/tattoo_Screenshot from 2025-02-07 14-53-43.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:15ea309c81057db659c9d660feeb25c60605bda2ef1e56264117c5365e8c87c8 +size 147073 diff --git a/dataset/query/american_flag46.jpg b/dataset/query/american_flag46.jpg new file mode 100644 index 0000000000000000000000000000000000000000..c8d78aecfba32962bdc9cb03aa50de3589b59598 Binary files /dev/null and b/dataset/query/american_flag46.jpg differ diff --git a/dataset/query/bird.jpg b/dataset/query/bird.jpg new file mode 100644 index 0000000000000000000000000000000000000000..f5cd8ed54e190d2106ba2553eb78d06d37d97392 Binary files /dev/null and b/dataset/query/bird.jpg differ diff --git a/dataset/query/british_flag40.jpg b/dataset/query/british_flag40.jpg new file mode 100644 index 0000000000000000000000000000000000000000..5bbeb557e1c55e2f358f6e18d415ab810d404452 Binary files /dev/null and b/dataset/query/british_flag40.jpg differ diff --git a/dataset/query/greek_flag20.jpg b/dataset/query/greek_flag20.jpg new file mode 100644 index 0000000000000000000000000000000000000000..fcc1b290d085fa07c0ba40c00d207b07c58475ae Binary files /dev/null and b/dataset/query/greek_flag20.jpg differ diff --git a/dataset/query/mandala.jpg b/dataset/query/mandala.jpg new file mode 100644 index 0000000000000000000000000000000000000000..6865c2e4956850e29a3b68f5d91520735fff0082 Binary files /dev/null and b/dataset/query/mandala.jpg differ diff --git a/dataset/query/moon.jpeg b/dataset/query/moon.jpeg new file mode 100644 index 0000000000000000000000000000000000000000..2aa062785965816b41c4c2afd1261a9cb7547893 Binary files /dev/null and b/dataset/query/moon.jpeg differ diff --git a/scripts/app.py b/scripts/app.py new file mode 100644 index 0000000000000000000000000000000000000000..2a4db5fd9f6eb39fdc396e754142853a75da9a03 --- /dev/null +++ b/scripts/app.py @@ -0,0 +1,99 @@ +import gradio as gr +import torch +from PIL import Image +import os +import numpy as np +import matplotlib.pyplot as plt +from transformers import CLIPProcessor, CLIPModel, AutoProcessor, AutoModel + +device = "cuda" if torch.cuda.is_available() else "cpu" + +# Load models and processors +clip_model = CLIPModel.from_pretrained("openai/clip-vit-base-patch32").to(device) +clip_processor = CLIPProcessor.from_pretrained("openai/clip-vit-base-patch32") +dino_model = AutoModel.from_pretrained("facebook/dinov2-base").to(device) +dino_processor = AutoProcessor.from_pretrained("facebook/dinov2-base") + +def get_image_embedding(image, model, processor, model_type): + if isinstance(image, str): # Handle file input + image = Image.open(image) + inputs = processor(images=image, return_tensors="pt").to(device) + with torch.no_grad(): + if model_type == "clip": + embedding = model.get_image_features(**inputs) + elif model_type == "dinov2": + outputs = model(**inputs) + embedding = outputs.last_hidden_state.mean(dim=1) # Global pooling + embedding /= embedding.norm(dim=-1, keepdim=True) # Normalize + return embedding + +def retrieve_images(query_img, gallery_imgs, model, processor, model_type, top_k=10): + query_embedding = get_image_embedding(query_img, model, processor, model_type) + + gallery_embeddings = [] + for img in gallery_imgs: + emb = get_image_embedding(img, model, processor, model_type) + gallery_embeddings.append((emb, img)) + + rank_list = [] + for emb, img in gallery_embeddings: + similarity_score = (query_embedding @ emb.T).item() + rank_list.append((similarity_score, img)) + + rank_list = sorted(rank_list, key=lambda x: x[0], reverse=True)[:top_k] + return [img for _, img in rank_list] + +def display_results(query_img, gallery_imgs, top_k): + clip_results = retrieve_images(query_img, gallery_imgs, clip_model, clip_processor, "clip", top_k) + dino_results = retrieve_images(query_img, gallery_imgs, dino_model, dino_processor, "dinov2", top_k) + return [query_img] + clip_results, [query_img] + dino_results + +def gradio_interface(query_img, gallery_imgs, top_k): + if not isinstance(gallery_imgs, list): + gallery_imgs = [gallery_imgs] + gallery_imgs = [img.name if hasattr(img, 'name') else img for img in gallery_imgs] # Handle NamedString issue + clip_res, dino_res = display_results(query_img, gallery_imgs, top_k) + return clip_res, dino_res + +import copy + +gallery_path = "dataset/gallery" +filenames = os.listdir(gallery_path) + +flag_filenames = [filename for filename in filenames if "flag" in filenames] +tattoo_filenames = [filename for filename in filenames if "tattoo" in filename] + +gallery_examples_flags = [os.path.join(gallery_path, filename) for filename in flag_filenames ] +gallery_examples_tattoos = [os.path.join(gallery_path, filename) for filename in tattoo_filenames ] + +query_examples = ["dataset/query/american_flag46.jpg", "dataset/query/bird.jpg"] + +print(gallery_examples_flags) +print(gallery_examples_tattoos) + +demo = gr.Interface( + fn=gradio_interface, + inputs=[ + gr.Image(type="pil", label="Query Image"), + gr.File(file_types=["image"], label="Gallery Images", file_count="multiple", elem_id="gallery-files"), + gr.Slider(1, 30, value=10, step=1, label="Top-K Matches"), + ], + outputs=[ + gr.Gallery(label="CLIP Retrieval Results", elem_id="clip-results", rows=[1], columns=[30]), + gr.Gallery(label="DINOv2 Retrieval Results", elem_id="dino-results", rows=[1], columns=[30]), + ], + title="CLIP vs DINOv2 Image Retrieval", + description="Upload a query image and gallery images to see the top-k retrieval results side by side using CLIP and DINOv2.", + examples=[[query_examples[0], gallery_examples_flags, 10], [query_examples[1], gallery_examples_tattoos, 10]], + css=""" + #gallery-files { + max-height: 150px; + overflow-y: scroll; + } + #clip-results, #dino-results { + max-height: 150px; + } + """ +) + +demo.launch(share=True) diff --git a/scripts/image_retrieval.py b/scripts/image_retrieval.py new file mode 100644 index 0000000000000000000000000000000000000000..483330f42e276f4d0c70cbdc6dfe95045755e71a --- /dev/null +++ b/scripts/image_retrieval.py @@ -0,0 +1,128 @@ +import torch +from PIL import Image +import os +from matplotlib import pyplot as plt +import argparse +from transformers import CLIPProcessor, CLIPModel +from transformers import AutoProcessor, AutoModel + +device = "cuda" if torch.cuda.is_available() else "cpu" + + +def create_gallery(gallery_paths, model, processor): + gallery = [] + + for path in gallery_paths: + img = Image.open(os.path.join(args.gallery_path,path)) + img_inputs = processor(images=img, return_tensors="pt").to(device) + with torch.no_grad(): + if args.model == "clip": + img_embedding = model.get_image_features(**img_inputs) + elif args.model == "dinov2": + with torch.no_grad(): + outputs = model(**img_inputs) + img_embedding = outputs.last_hidden_state.mean(dim=1) + img_embedding /= img_embedding.norm(dim=-1, keepdim=True) + + gallery.append([img_embedding, os.path.join(args.gallery_path, path)]) + + return gallery + +def retrieval(args): + if args.model == "clip": + model = CLIPModel.from_pretrained("openai/clip-vit-base-patch32").to(device) + processor = CLIPProcessor.from_pretrained("openai/clip-vit-base-patch32") + elif args.model == "dinov2": + # Load DINOv2 model + model_name = "facebook/dinov2-base" + model = AutoModel.from_pretrained(model_name) + processor = AutoProcessor.from_pretrained(model_name) + + + gallery_paths = os.listdir(args.gallery_path) + query_paths = os.listdir(args.query_path) + + print("--- Initalizing gallery ---") + gallery = create_gallery(gallery_paths, model, processor) + + for k, query_path in enumerate(query_paths): + query_image = Image.open(os.path.join(args.query_path, query_path)) + img_inputs = processor(images=query_image, return_tensors="pt").to(device) + with torch.no_grad(): + if args.model == "clip": + query_embedding = model.get_image_features(**img_inputs) + elif args.model == "dinov2": + with torch.no_grad(): + outputs = model(**img_inputs) + query_embedding = outputs.last_hidden_state.mean(dim=1) + query_embedding /= query_embedding.norm(dim=-1, keepdim=True) + + + fig = plt.figure() + plot_length = 11 + rank_list = [] + + gallery_ax = fig.add_subplot(1,plot_length,1) #add query image in the left top place in plot + gallery_ax.imshow(query_image) + + print(f"--- Starting image retrieval for query image: {query_path}") + logit_scale = 100 + query_normalized = query_embedding / query_embedding.norm(dim=1, keepdim=True) + + for item in gallery: + # normalized features + gallery_normalized = item[0] / item[0].norm(dim=1, keepdim=True) + # cosine similarity as logits + similarity_score = (logit_scale * query_normalized @ gallery_normalized.t()).item() + similarity_score = round(similarity_score,3) + rank_list.append([similarity_score, item[1]]) # add gallery image with its similarity score to this query image in ranking list + + rank_list = sorted(rank_list, key=lambda x: x[0], reverse = True) + + + for i in range(2,plot_length): + gallery_ax = fig.add_subplot(1,plot_length,i) + img = Image.open(rank_list[i][1]) + gallery_ax.imshow(img) + gallery_ax.set_title('%.1f'% rank_list[i][0], fontsize=8) #add similarity score as title + gallery_ax.axis('off') + plt.savefig(os.path.join(args.outDir, "plot_"+ str(k)+".jpg")) + plt.close() + + + +if __name__ == "__main__": + # Create an argument parser + parser = argparse.ArgumentParser(description="CLIP Image Retriever") + + # Add arguments + parser.add_argument( + '--gallery-path', + type=str, + default="dataset/gallery/", + help="Directory containing the gallery images" + ) + parser.add_argument( + '--query-path', + type=str, + default="dataset/query/", + help="Directory containing the query images" + ) + parser.add_argument( + '--outDir', + type=str, + default="outputs/retrieval_clip", + help="Directory containing the output plots" + ) + parser.add_argument( + '--model', + type=str, + default="clip", + help="Model type. i.e clip or dinov2" + ) + + # Parse the arguments + args = parser.parse_args() + + os.makedirs(args.outDir, exist_ok=True) + retrieval(args) \ No newline at end of file