Pre-trained models

December 7, 2023 · View on GitHub

The following table summarizes the performance of the pre-trained models and the URLs of checkpoints and configurations. The scores differ from the paper due to refactoring of the code (especially AR, which improves performance substantially). We set --seed_everything 0 for both training and evaluation. The training logs can be found at TensorBoard.dev here.

NameAccuracy (RGB)Accuracy (Alpha)F-score (RGB)F-score (Alpha)BG-FCDText-FCDPixel-FCDContrast (% Pages)Contrast (# Elements)URLNote
Stats (mode)0.7170.8910.0030.21922.15265.087.4923.790.59ckpt, configEvaluate with --model.sampling false.
Stats (sampling)0.6200.8210.0040.2060.7183.55172.9493.354.07ckpt, configEvaluate with --model.sampling true.
AR (greedy)0.7580.9160.0640.6122.1423.9112.6365.691.96ckpt, configEvaluate with --model.top_p 0.0.
AR (top-p, p=0.8)0.7550.9150.0620.6051.9822.3318.5768.632.11ckpt, configEvaluate with --model.top_p 0.8.
AR (top-p, p=0.9)0.7530.9140.0610.6041.8421.0021.3369.782.19ckpt, configEvaluate with --model.top_p 0.9.
NAR0.7740.9290.0780.6771.9722.467.8071.832.22ckpt, config
CVAE0.7730.9290.0690.6631.6528.677.9072.912.23ckpt, config
Upsampler---------ckpt, configUsed together with all the other models.