Ricercar commited on
Commit
49e2601
1 Parent(s): 666f40e

update crop and resize comparison

Browse files
.gitignore CHANGED
@@ -3,7 +3,9 @@ ehthumbs.db
3
  Icon?
4
  Thumbs.db
5
  *.DS_Store
 
6
 
7
  secret.sh
8
 
9
- .idea/
 
 
3
  Icon?
4
  Thumbs.db
5
  *.DS_Store
6
+ *.pyc
7
 
8
  secret.sh
9
 
10
+ .idea/
11
+
data/promptbook/data-00000-of-00021.arrow CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:9313ff854cf203b11d4ee57c49337daaab1bc1aa137def11a80a2f22247387c8
3
- size 496158744
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3e74589073da0de6b37f223ebe366f0e16fbd4fe6fddbf56c306e7503a4ba1f8
3
+ size 496186912
data/promptbook/data-00001-of-00021.arrow CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:73805e837a90d1878648a43104d73328b553326ea1f00019c954b67ba04ab9ed
3
- size 495185576
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:09e453323562f04ab085ddf3a5a3dcca809d0c629733bebef9fcfdca7be53833
3
+ size 495213744
data/promptbook/data-00002-of-00021.arrow CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:10252d2c45ea1ba88f3b5e0f3bb0e2b08099606036e6f3638e8449faff36057e
3
- size 495550496
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f8d4ace0e12d35036cc24c822abc3ff20417a1fdca93fcf4277ba726c6a518bc
3
+ size 495578664
data/promptbook/data-00003-of-00021.arrow CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:7ff11c7c98b73f2c1824d6f98ea09d8789e6909a728eb39420caa3b8333b2af7
3
- size 495623208
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ba447c753a2cca7d3462fd7af4bb997de39a42a44b226ceb63f700acf63a9f59
3
+ size 495651344
data/promptbook/data-00004-of-00021.arrow CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:4ed29ad5f10230a5f868abd32684db99bb53540036b59cec305f8266b20c48bd
3
- size 494484472
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7f1e3cf3b0308795780c2fa3fb0ffa99fc6f7ba176bc9e6f2e240b5f80fd37ca
3
+ size 494512608
data/promptbook/data-00005-of-00021.arrow CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:d70104e55ac2752f6dbf658f3619d4dfb55cf9bf958dd422b947bc13370e36c2
3
- size 492936800
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:beb3634bc09c153b1d2e6b8393f97d11d9fbfd48c72fd31a18b7523c164b97f1
3
+ size 492964936
data/promptbook/data-00006-of-00021.arrow CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:37a485ae5318e30569a98a968115f8907ad0d3e21e1b0f94562c2bde82042e7b
3
- size 490180944
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:22d2a81538bb62affec34c14955fac0f84a10d6b7eb55f43367a8fa0f90a1387
3
+ size 490209080
data/promptbook/data-00007-of-00021.arrow CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:dff6f5876530d08a62b0af3b0ff7cce7a141afcb4789cd5aa973ef46ed443680
3
- size 490213760
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4e971eaaa13ec11b565bd62d6d61d9e572e0b8ea7e066917756d87fec3d8c423
3
+ size 490241896
data/promptbook/data-00008-of-00021.arrow CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:6c5de841dd2fa29c29acb7b0cfdf9334c61b6d5654bf64ecdd56f0bb5959c956
3
- size 492484024
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:67b1b560a03d11b1c6829c708f94bc71dd3f8064f0d126c4b4dc84cf4bf630db
3
+ size 492512160
data/promptbook/data-00009-of-00021.arrow CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:de5d0a7132b3befd2c1a94f2ce870d02dffab1c440e65d277e58dc8b40563bf4
3
- size 494710624
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a09cb25b4b67484b248ed56b7f1e2aec3fd4ce8559b42ff6c8ec6b2ac156e8f0
3
+ size 494738760
data/promptbook/data-00010-of-00021.arrow CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:57cfd4def42baf69d49439021a6d49465df4a2cd2ddb2d8c22170598710f3ea8
3
- size 497841824
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a10c5ad9ae0a021e9adc194a409744e2033c40364efe28cf2129056e2d2655b4
3
+ size 497869960
data/promptbook/data-00011-of-00021.arrow CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:35d6f95a91c711453b12f4abf6f67f3bd8cb4b46d02b827f8e62ae0b3c83e4b6
3
- size 494884984
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:416f48e2623c5a936a800fdc31d2c8d0d3166199af401c1f90a9e1211e4a32f5
3
+ size 494913120
data/promptbook/data-00012-of-00021.arrow CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:dc94e7858ca01950b0f5a4416a8665e33a9d2428256a81883ef2eadb9458dd3f
3
- size 488987880
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6791af13fea5e3fae2243ae1109d6ed5826821af613ba45948fa4d49e682434e
3
+ size 489016016
data/promptbook/data-00013-of-00021.arrow CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:fcfa4e959819b4766f6085acdb5087be18fbb59e752b9497de7aed1e9eede057
3
- size 498320888
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bf7b3526921f797af2ece5c1b9491fe5cab713df982f3edf90f8857bd284f0c4
3
+ size 498349024
data/promptbook/data-00014-of-00021.arrow CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:560c856d65954946f2746b160a7698eac9f01928f8e18d2b126850797fdffe93
3
- size 489513032
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0972ffc62d8f7d7c772ee1006876de444640122fa967ce30bef67aed3d6acbd8
3
+ size 489541168
data/promptbook/data-00015-of-00021.arrow CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:31b905f1c84f47005543cbfb4178a0af13d5e066d34a2bc71e3e94823861ed12
3
- size 495904920
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f95b794b449aa711fca297afad2989fe12fe38ca1b06ad92c4c1f3392fcec6bd
3
+ size 495933056
data/promptbook/data-00016-of-00021.arrow CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:ab14fdd99b4e3c1a56ff786bfa24d4c679b0973f5d60e93cadf640f03934109d
3
- size 490964960
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9f21b6e94354ac68e7cf4ca3065704cb639d27c21402d0e93a97171b2cd08a80
3
+ size 490993096
data/promptbook/data-00017-of-00021.arrow CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:6a63fa98450893c64027c059a72cb0ef349f237fdb8b6057801899422a1e7f8f
3
- size 499506568
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:018f0507e4c61140487f1e5ec8460cd639d2a369ef38e78a4aba7d19cc519e5c
3
+ size 499534704
data/promptbook/data-00018-of-00021.arrow CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:6b497f822200a4adc68c2cfb6e92f8de67335d92b3b8b5b7c5dc19f207ec2b79
3
- size 493819464
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bec6c864be6674477e83e0250adecde9ae4289a3ad257f2ae761944932407c80
3
+ size 493847600
data/promptbook/data-00019-of-00021.arrow CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:1d1c4f49a5fc2f04c9a519669504da3e5462e3bffbc3208c9ec66bb9a11d616b
3
- size 492267312
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:09a5489c77c503143698e2382f1a9d5e5bb712d42c17d6ef45bf5d95fdde85e9
3
+ size 492295448
data/promptbook/data-00020-of-00021.arrow CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:32132052c4f7a2631f3180364d978db9bdc49bd23cfa36493ec42a646960a770
3
- size 492600960
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9012878da88419099e86db7fd0aa20be5d91ca2c0ac816601fc40398505431c3
3
+ size 492629096
data/promptbook/dataset_info.json CHANGED
@@ -1,94 +1,94 @@
1
  {
2
  "citation": "",
3
- "dataset_size": 10372092350,
4
  "description": "",
5
  "download_checksums": {
6
- "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/5944fa3c00a863a530b6d74819d9108dac3b0af9/data/train-00000-of-00021-dce1ab300958e31f.parquet": {
7
- "num_bytes": 496016461,
8
  "checksum": null
9
  },
10
- "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/5944fa3c00a863a530b6d74819d9108dac3b0af9/data/train-00001-of-00021-bc9644c1d3743699.parquet": {
11
- "num_bytes": 495045000,
12
  "checksum": null
13
  },
14
- "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/5944fa3c00a863a530b6d74819d9108dac3b0af9/data/train-00002-of-00021-ea2e66702961212b.parquet": {
15
- "num_bytes": 495409902,
16
  "checksum": null
17
  },
18
- "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/5944fa3c00a863a530b6d74819d9108dac3b0af9/data/train-00003-of-00021-a9bd0a2782c2c69d.parquet": {
19
- "num_bytes": 495478259,
20
  "checksum": null
21
  },
22
- "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/5944fa3c00a863a530b6d74819d9108dac3b0af9/data/train-00004-of-00021-90c88f250e8f6b62.parquet": {
23
- "num_bytes": 494335691,
24
  "checksum": null
25
  },
26
- "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/5944fa3c00a863a530b6d74819d9108dac3b0af9/data/train-00005-of-00021-b2c5536485ce34b3.parquet": {
27
- "num_bytes": 492796254,
28
  "checksum": null
29
  },
30
- "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/5944fa3c00a863a530b6d74819d9108dac3b0af9/data/train-00006-of-00021-971695ae84e53afe.parquet": {
31
- "num_bytes": 490037891,
32
  "checksum": null
33
  },
34
- "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/5944fa3c00a863a530b6d74819d9108dac3b0af9/data/train-00007-of-00021-3b1e0c653bc45819.parquet": {
35
- "num_bytes": 490068792,
36
  "checksum": null
37
  },
38
- "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/5944fa3c00a863a530b6d74819d9108dac3b0af9/data/train-00008-of-00021-e99a99266f821af8.parquet": {
39
- "num_bytes": 492340436,
40
  "checksum": null
41
  },
42
- "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/5944fa3c00a863a530b6d74819d9108dac3b0af9/data/train-00009-of-00021-16fa05c06e124b7d.parquet": {
43
- "num_bytes": 494567838,
44
  "checksum": null
45
  },
46
- "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/5944fa3c00a863a530b6d74819d9108dac3b0af9/data/train-00010-of-00021-9f81d2c90824a45d.parquet": {
47
- "num_bytes": 497703057,
48
  "checksum": null
49
  },
50
- "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/5944fa3c00a863a530b6d74819d9108dac3b0af9/data/train-00011-of-00021-b1abb8fe24b20a1d.parquet": {
51
- "num_bytes": 494741057,
52
  "checksum": null
53
  },
54
- "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/5944fa3c00a863a530b6d74819d9108dac3b0af9/data/train-00012-of-00021-d67c472e42578ac1.parquet": {
55
- "num_bytes": 488841048,
56
  "checksum": null
57
  },
58
- "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/5944fa3c00a863a530b6d74819d9108dac3b0af9/data/train-00013-of-00021-27e4724254d903b4.parquet": {
59
- "num_bytes": 498179762,
60
  "checksum": null
61
  },
62
- "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/5944fa3c00a863a530b6d74819d9108dac3b0af9/data/train-00014-of-00021-72f0d1125f62ff50.parquet": {
63
- "num_bytes": 489363694,
64
  "checksum": null
65
  },
66
- "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/5944fa3c00a863a530b6d74819d9108dac3b0af9/data/train-00015-of-00021-66562f62d5cfd2d7.parquet": {
67
- "num_bytes": 495756794,
68
  "checksum": null
69
  },
70
- "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/5944fa3c00a863a530b6d74819d9108dac3b0af9/data/train-00016-of-00021-741d2101a8ac38d6.parquet": {
71
- "num_bytes": 490821807,
72
  "checksum": null
73
  },
74
- "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/5944fa3c00a863a530b6d74819d9108dac3b0af9/data/train-00017-of-00021-30d22f60bedc5a0a.parquet": {
75
- "num_bytes": 499362634,
76
  "checksum": null
77
  },
78
- "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/5944fa3c00a863a530b6d74819d9108dac3b0af9/data/train-00018-of-00021-d553d608301c44e2.parquet": {
79
- "num_bytes": 493676628,
80
  "checksum": null
81
  },
82
- "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/5944fa3c00a863a530b6d74819d9108dac3b0af9/data/train-00019-of-00021-7d5ce4f46e18cac7.parquet": {
83
- "num_bytes": 492125301,
84
  "checksum": null
85
  },
86
- "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/5944fa3c00a863a530b6d74819d9108dac3b0af9/data/train-00020-of-00021-7f5789631bfc4dd7.parquet": {
87
- "num_bytes": 492454258,
88
  "checksum": null
89
  }
90
  },
91
- "download_size": 10369122564,
92
  "features": {
93
  "image": {
94
  "_type": "Image"
@@ -141,19 +141,23 @@
141
  "dtype": "string",
142
  "_type": "Value"
143
  },
144
- "clip_score": {
145
  "dtype": "float64",
146
  "_type": "Value"
147
  },
148
- "msq_score": {
149
  "dtype": "float64",
150
  "_type": "Value"
151
  },
152
- "nsfw_score": {
153
  "dtype": "float64",
154
  "_type": "Value"
155
  },
156
- "norm_clip": {
 
 
 
 
157
  "dtype": "float64",
158
  "_type": "Value"
159
  },
@@ -165,18 +169,30 @@
165
  "dtype": "float64",
166
  "_type": "Value"
167
  },
168
- "norm_msq": {
 
 
 
 
 
 
 
 
 
 
 
 
169
  "dtype": "float64",
170
  "_type": "Value"
171
  }
172
  },
173
  "homepage": "",
174
  "license": "",
175
- "size_in_bytes": 20741214914,
176
  "splits": {
177
  "train": {
178
  "name": "train",
179
- "num_bytes": 10372092350,
180
  "num_examples": 18000,
181
  "shard_lengths": [
182
  1716,
 
1
  {
2
  "citation": "",
3
+ "dataset_size": 10372677422,
4
  "description": "",
5
  "download_checksums": {
6
+ "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/c40f13b1561dcd77c80594d4b82c84c2a0e05c82/data/train-00000-of-00021-144de512f198a571.parquet": {
7
+ "num_bytes": 496060014,
8
  "checksum": null
9
  },
10
+ "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/c40f13b1561dcd77c80594d4b82c84c2a0e05c82/data/train-00001-of-00021-86e3cd5e72c66fc3.parquet": {
11
+ "num_bytes": 495088216,
12
  "checksum": null
13
  },
14
+ "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/c40f13b1561dcd77c80594d4b82c84c2a0e05c82/data/train-00002-of-00021-5ee5b816ac97f84d.parquet": {
15
+ "num_bytes": 495453484,
16
  "checksum": null
17
  },
18
+ "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/c40f13b1561dcd77c80594d4b82c84c2a0e05c82/data/train-00003-of-00021-d82381034c0d0481.parquet": {
19
+ "num_bytes": 495521595,
20
  "checksum": null
21
  },
22
+ "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/c40f13b1561dcd77c80594d4b82c84c2a0e05c82/data/train-00004-of-00021-c547926606ead515.parquet": {
23
+ "num_bytes": 494379154,
24
  "checksum": null
25
  },
26
+ "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/c40f13b1561dcd77c80594d4b82c84c2a0e05c82/data/train-00005-of-00021-b24c5056a962ab0e.parquet": {
27
+ "num_bytes": 492839660,
28
  "checksum": null
29
  },
30
+ "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/c40f13b1561dcd77c80594d4b82c84c2a0e05c82/data/train-00006-of-00021-78049573f1481e30.parquet": {
31
+ "num_bytes": 490081012,
32
  "checksum": null
33
  },
34
+ "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/c40f13b1561dcd77c80594d4b82c84c2a0e05c82/data/train-00007-of-00021-b826900e0fd2dce7.parquet": {
35
+ "num_bytes": 490112233,
36
  "checksum": null
37
  },
38
+ "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/c40f13b1561dcd77c80594d4b82c84c2a0e05c82/data/train-00008-of-00021-732145cd38c9c168.parquet": {
39
+ "num_bytes": 492383717,
40
  "checksum": null
41
  },
42
+ "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/c40f13b1561dcd77c80594d4b82c84c2a0e05c82/data/train-00009-of-00021-c6885fc6c87dbc38.parquet": {
43
+ "num_bytes": 494611087,
44
  "checksum": null
45
  },
46
+ "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/c40f13b1561dcd77c80594d4b82c84c2a0e05c82/data/train-00010-of-00021-9218905547d91424.parquet": {
47
+ "num_bytes": 497746263,
48
  "checksum": null
49
  },
50
+ "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/c40f13b1561dcd77c80594d4b82c84c2a0e05c82/data/train-00011-of-00021-b2de8d961120a62f.parquet": {
51
+ "num_bytes": 494784264,
52
  "checksum": null
53
  },
54
+ "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/c40f13b1561dcd77c80594d4b82c84c2a0e05c82/data/train-00012-of-00021-7494a4330b966941.parquet": {
55
+ "num_bytes": 488884400,
56
  "checksum": null
57
  },
58
+ "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/c40f13b1561dcd77c80594d4b82c84c2a0e05c82/data/train-00013-of-00021-837562090b2a7997.parquet": {
59
+ "num_bytes": 498223065,
60
  "checksum": null
61
  },
62
+ "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/c40f13b1561dcd77c80594d4b82c84c2a0e05c82/data/train-00014-of-00021-6262400e976d9754.parquet": {
63
+ "num_bytes": 489407108,
64
  "checksum": null
65
  },
66
+ "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/c40f13b1561dcd77c80594d4b82c84c2a0e05c82/data/train-00015-of-00021-9e46942c1683f62e.parquet": {
67
+ "num_bytes": 495800178,
68
  "checksum": null
69
  },
70
+ "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/c40f13b1561dcd77c80594d4b82c84c2a0e05c82/data/train-00016-of-00021-6da442e4f4b5179c.parquet": {
71
+ "num_bytes": 490865086,
72
  "checksum": null
73
  },
74
+ "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/c40f13b1561dcd77c80594d4b82c84c2a0e05c82/data/train-00017-of-00021-6dbed53bcdcf3e27.parquet": {
75
+ "num_bytes": 499405883,
76
  "checksum": null
77
  },
78
+ "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/c40f13b1561dcd77c80594d4b82c84c2a0e05c82/data/train-00018-of-00021-05417d33e030566b.parquet": {
79
+ "num_bytes": 493719879,
80
  "checksum": null
81
  },
82
+ "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/c40f13b1561dcd77c80594d4b82c84c2a0e05c82/data/train-00019-of-00021-e123aec9df97bb47.parquet": {
83
+ "num_bytes": 492168855,
84
  "checksum": null
85
  },
86
+ "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/c40f13b1561dcd77c80594d4b82c84c2a0e05c82/data/train-00020-of-00021-aa7cbe378939fd0a.parquet": {
87
+ "num_bytes": 492497529,
88
  "checksum": null
89
  }
90
  },
91
+ "download_size": 10370032682,
92
  "features": {
93
  "image": {
94
  "_type": "Image"
 
141
  "dtype": "string",
142
  "_type": "Value"
143
  },
144
+ "nsfw_score": {
145
  "dtype": "float64",
146
  "_type": "Value"
147
  },
148
+ "mcos_score_crop": {
149
  "dtype": "float64",
150
  "_type": "Value"
151
  },
152
+ "clip_score_crop": {
153
  "dtype": "float64",
154
  "_type": "Value"
155
  },
156
+ "norm_clip_crop": {
157
+ "dtype": "float64",
158
+ "_type": "Value"
159
+ },
160
+ "norm_mcos_crop": {
161
  "dtype": "float64",
162
  "_type": "Value"
163
  },
 
169
  "dtype": "float64",
170
  "_type": "Value"
171
  },
172
+ "mcos_score_resize": {
173
+ "dtype": "float64",
174
+ "_type": "Value"
175
+ },
176
+ "clip_score_resize": {
177
+ "dtype": "float64",
178
+ "_type": "Value"
179
+ },
180
+ "norm_clip_resize": {
181
+ "dtype": "float64",
182
+ "_type": "Value"
183
+ },
184
+ "norm_mcos_resize": {
185
  "dtype": "float64",
186
  "_type": "Value"
187
  }
188
  },
189
  "homepage": "",
190
  "license": "",
191
+ "size_in_bytes": 20742710104,
192
  "splits": {
193
  "train": {
194
  "name": "train",
195
+ "num_bytes": 10372677422,
196
  "num_examples": 18000,
197
  "shard_lengths": [
198
  1716,
data/promptbook/state.json CHANGED
@@ -64,7 +64,7 @@
64
  "filename": "data-00020-of-00021.arrow"
65
  }
66
  ],
67
- "_fingerprint": "43ce89820094f517",
68
  "_format_columns": null,
69
  "_format_kwargs": {},
70
  "_format_type": null,
 
64
  "filename": "data-00020-of-00021.arrow"
65
  }
66
  ],
67
+ "_fingerprint": "312c6d8087aef10f",
68
  "_format_columns": null,
69
  "_format_kwargs": {},
70
  "_format_type": null,
pages/Gallery.py CHANGED
@@ -82,6 +82,9 @@ class GalleryApp:
82
  st.write(f"**{key}**: {items.iloc[idx + j][key]}")
83
 
84
  def selection_panel(self, items):
 
 
 
85
  selecters = st.columns([1, 4])
86
 
87
  if 'score_weights' not in st.session_state:
@@ -112,18 +115,18 @@ class GalleryApp:
112
  with sub_selecters[0]:
113
  clip_weight = st.number_input('Clip Score Weight', min_value=-100.0, max_value=100.0, value=st.session_state.score_weights[0], step=0.1, help='the weight for normalized clip score')
114
  with sub_selecters[1]:
115
- msq_weight = st.number_input('mSQ Weight', min_value=-100.0, max_value=100.0, value=st.session_state.score_weights[1], step=0.1, help='the weight for m(eam) s(imilarity) q(antile) score for measuring distinctiveness')
116
  with sub_selecters[2]:
117
  pop_weight = st.number_input('Popularity Weight', min_value=-100.0, max_value=100.0, value=st.session_state.score_weights[2], step=0.1, help='the weight for normalized popularity score')
118
 
119
- items.loc[:, 'weighted_score_sum'] = round(items['norm_clip'] * clip_weight + items['norm_msq'] * msq_weight + items[
120
  'norm_pop'] * pop_weight, 4)
121
 
122
  continue_idx = 3
123
 
124
  # save latest weights
125
  st.session_state.score_weights[0] = clip_weight
126
- st.session_state.score_weights[1] = msq_weight
127
  st.session_state.score_weights[2] = pop_weight
128
 
129
  # select threshold
@@ -173,7 +176,7 @@ class GalleryApp:
173
  # select info to show
174
  info = st.multiselect('Show Info',
175
  ['model_name', 'model_id', 'modelVersion_name', 'modelVersion_id',
176
- 'weighted_score_sum', 'model_download_count', 'clip_score', 'msq_score',
177
  'nsfw_score', 'norm_nsfw'],
178
  default=sort_by)
179
 
@@ -290,7 +293,6 @@ class GalleryApp:
290
  with st.spinner('Loading images...'):
291
  self.gallery_standard(items, col_num, info)
292
 
293
-
294
  def submit_actions(self, status, prompt_id):
295
  if status == 'Select':
296
  modelVersions = self.promptBook[self.promptBook['prompt_id'] == prompt_id]['modelVersion_id'].unique()
 
82
  st.write(f"**{key}**: {items.iloc[idx + j][key]}")
83
 
84
  def selection_panel(self, items):
85
+ # temperal function
86
+ preprocessor = st.radio('Preprocess Method', ['crop', 'resize'], horizontal=True)
87
+
88
  selecters = st.columns([1, 4])
89
 
90
  if 'score_weights' not in st.session_state:
 
115
  with sub_selecters[0]:
116
  clip_weight = st.number_input('Clip Score Weight', min_value=-100.0, max_value=100.0, value=st.session_state.score_weights[0], step=0.1, help='the weight for normalized clip score')
117
  with sub_selecters[1]:
118
+ mcos_weight = st.number_input('Dissimilarity Weight', min_value=-100.0, max_value=100.0, value=st.session_state.score_weights[1], step=0.1, help='the weight for m(eam) s(imilarity) q(antile) score for measuring distinctiveness')
119
  with sub_selecters[2]:
120
  pop_weight = st.number_input('Popularity Weight', min_value=-100.0, max_value=100.0, value=st.session_state.score_weights[2], step=0.1, help='the weight for normalized popularity score')
121
 
122
+ items.loc[:, 'weighted_score_sum'] = round(items[f'norm_clip_{preprocessor}'] * clip_weight + items[f'norm_mcos_{preprocessor}'] * mcos_weight + items[
123
  'norm_pop'] * pop_weight, 4)
124
 
125
  continue_idx = 3
126
 
127
  # save latest weights
128
  st.session_state.score_weights[0] = clip_weight
129
+ st.session_state.score_weights[1] = mcos_weight
130
  st.session_state.score_weights[2] = pop_weight
131
 
132
  # select threshold
 
176
  # select info to show
177
  info = st.multiselect('Show Info',
178
  ['model_name', 'model_id', 'modelVersion_name', 'modelVersion_id',
179
+ 'weighted_score_sum', 'model_download_count', 'clip_score', 'mcos_score',
180
  'nsfw_score', 'norm_nsfw'],
181
  default=sort_by)
182
 
 
293
  with st.spinner('Loading images...'):
294
  self.gallery_standard(items, col_num, info)
295
 
 
296
  def submit_actions(self, status, prompt_id):
297
  if status == 'Select':
298
  modelVersions = self.promptBook[self.promptBook['prompt_id'] == prompt_id]['modelVersion_id'].unique()
pages/__pycache__/Gallery.cpython-39.pyc CHANGED
Binary files a/pages/__pycache__/Gallery.cpython-39.pyc and b/pages/__pycache__/Gallery.cpython-39.pyc differ