multi_vae_comparison_mega_liminal.json

{
  "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
      ]
    ]
  }
}