# GenerateCT: Text-Guided 3D Chest CT Generation
Welcome to the official page of GenerateCT, a pioneering work in text-conditional 3D medical image generation with a particular focus on chest CT volumes. Here, you will find the code, all trained models for text-to-CT generation, and a dataset of generated CT volumes paired with text prompts, all freely accessible for researchers.

<p align="center">
  <img src="https://huggingface.co/generatect/GenerateCT/resolve/main/figures/examples.gif" width="100%">
</p>

## Results
Our models achieve the following performances:

<table>
  <thead>
    <tr style="text-align: center; background-color: #f8f8f8;">
      <th>Model</th>
      <th>Resolution (px)</th>
      <th>Dimension</th>
      <th>Text-Guided?</th>
      <th>FID Score (↓)</th>
      <th>FVD Score (↓)</th>
      <th>CLIP Score (↑)</th>
    </tr>
  </thead>
  <tbody>
    <tr style="text-align: center;">
      <td>CT-ViT</td>
      <td>128</td>
      <td>3D</td>
      <td>No</td>
      <td>73.4</td>
      <td>1817.4</td>
      <td>N/A</td>
    </tr>
    <tr style="text-align: center;">
      <td>Transformer</td>
      <td>128</td>
      <td>3D</td>
      <td>Yes</td>
      <td>104.3</td>
      <td>1886.8</td>
      <td>25.2</td>
    </tr>
    <tr style="text-align: center;">
      <td>Diffusion</td>
      <td>512</td>
      <td>2D</td>
      <td>Yes</td>
      <td>14.9</td>
      <td>409.8</td>
      <td>27.6</td>
    </tr>
    <tr style="font-weight: bold; text-align: center;">
      <td>GenerateCT</td>
      <td>512</td>
      <td>3D</td>
      <td>Yes</td>
      <td>55.8</td>
      <td>1092.3</td>
      <td>27.1</td>
    </tr>
  </tbody>
</table>

## License
Our work, including the codes, trained models, and generated data, is released under a [Creative Commons Attribution (CC-BY) license](https://creativecommons.org/licenses/by/4.0/). This means that anyone is free to share (copy and redistribute the material in any medium or format) and adapt (remix, transform, and build upon the material) for any purpose, even commercially, as long as appropriate credit is given, a link to the license is provided, and any changes that were made are indicated. This aligns with our goal of facilitating progress in the field by providing a resource for researchers to build upon.