ahmedgh970 / brain-anomaly-seg

Transformer-based Models for Unsupervised Anomaly Segmentation in Brain MR Images
GNU General Public License v3.0
19 stars 3 forks source link

VQ-VAE with TF #2

Closed hibiat closed 2 years ago

hibiat commented 2 years ago

Just wondering if the code of "VQ-VAE + Transformer" in your paper has been uploaded. Is it H_TAE.py or anything else? Thanks!

ahmedgh970 commented 2 years ago

The "VQ-VAE + Transformer" model was proposed in Pinaya's work https://openreview.net/pdf?id=Z1tlNqbCpp_ and we just report their results on BraTS and MSLUB datasets. But keep in mind that their models have not been trained on the same training set as the benchmark and our proposed models.