{ "SD 1.5": { "latent_shape": [ 4, 64, 64 ], "n_images": 2074, "total_annotations": 229618, "class_counts": { "pentachoron": 80065, "line_diag": 4241, "saddle": 129783, "tetrahedron": 2564, "cone": 3630, "triangular_prism": 6523, "tube": 1496, "plane": 875, "square_xy": 237, "torus": 109, "cylinder": 52, "capsule": 16, "hemisphere": 2, "octahedron": 2, "shell": 12, "cube": 5, "sphere": 2, "cuboid": 4 }, "mean_confidence": 0.8804939533307081, "scales": [ [ 4, 64, 64 ], [ 4, 32, 32 ], [ 4, 16, 16 ] ], "channel_groups": [ [ 0 ], [ 2 ], [ 3 ], [ 1 ] ] }, "SDXL": { "latent_shape": [ 4, 128, 128 ], "n_images": 2074, "total_annotations": 1092657, "class_counts": { "line_diag": 44622, "saddle": 574252, "pentachoron": 332100, "tetrahedron": 15745, "triangular_prism": 66206, "square_xy": 17893, "plane": 31678, "tube": 2818, "cone": 6516, "torus": 138, "cylinder": 527, "disc": 55, "shell": 73, "cube": 8, "helix": 5, "hemisphere": 10, "sphere": 4, "capsule": 5, "octahedron": 2 }, "mean_confidence": 0.8735912484119882, "scales": [ [ 4, 64, 64 ], [ 4, 32, 32 ], [ 4, 16, 16 ] ], "channel_groups": [ [ 0 ], [ 2 ], [ 3 ], [ 1 ] ] }, "Flux.1": { "latent_shape": [ 16, 128, 128 ], "n_images": 2074, "total_annotations": 2943743, "class_counts": { "pentachoron": 906158, "triangular_prism": 207696, "square_xy": 432908, "plane": 841923, "saddle": 433827, "tetrahedron": 73956, "tube": 22120, "cuboid": 4016, "cone": 3636, "cube": 170, "torus": 1122, "rectangle": 192, "line_diag": 13326, "disc": 1378, "cylinder": 948, "capsule": 178, "ellipse": 127, "cross": 14, "sphere": 14, "shell": 18, "hemisphere": 6, "bowl": 4, "octahedron": 4, "pyramid": 2 }, "mean_confidence": 0.8782098689124771, "scales": [ [ 16, 64, 64 ], [ 8, 32, 32 ], [ 8, 16, 16 ] ], "channel_groups": [ [ 0, 6 ], [ 5, 2 ], [ 4, 9 ], [ 3, 7 ], [ 8, 10 ], [ 11, 13 ], [ 15, 1 ], [ 14, 12 ] ] }, "Flux.2": { "latent_shape": [ 32, 128, 128 ], "n_images": 2074, "total_annotations": 4365328, "class_counts": { "line_diag": 125249, "tetrahedron": 162493, "pentachoron": 932633, "saddle": 3040696, "triangular_prism": 56747, "square_xy": 8209, "plane": 35631, "tube": 2338, "cone": 834, "torus": 43, "cuboid": 240, "cylinder": 117, "disc": 83, "cube": 9, "ellipse": 2, "shell": 2, "sphere": 2 }, "mean_confidence": 0.8750485207082765, "scales": [ [ 16, 64, 64 ], [ 8, 32, 32 ], [ 8, 16, 16 ] ], "channel_groups": [ [ 0, 20, 9, 18 ], [ 11, 2, 8, 19 ], [ 13, 30, 14, 12 ], [ 6, 7, 3, 5 ], [ 4, 17, 26, 24 ], [ 10, 25, 21, 22 ], [ 29, 15, 23, 1 ], [ 16, 31, 28, 27 ] ] } }