Cool-chic code at a glanceΒΆ

Coolchic source code is organized as follows

coolchic/
    |___ enc/
    |     |___ component/                     The different modules composing the Encoder
    |     |          |___ core/               The different NN (ARM, upsampling and synthesis) + quantizer
    |     |          |___ coolchic.py         Entire Cool-chic: Latent + ARM + upsampling + synthesis
    |     |          |___ frame.py            Encode a frame with a cool-chic
    |     |          |___ video.py            Encode a video. Call frame.py for each frame
    |     |___ io/                            Read & write PNG, PPM and YUV
    |     |___ training/                      Train, test, quantize the models, loss, presets, warm-up
    |     |___ utils/                         Coding structure and various stuff.
    |     |___ bitstream/                     Write the bitstream from a trained model.
    |
    |___ dec/                                 Python interface with the C API for the decoder
    |
    |___ cpp/                                 Cool-chic C API for fast decoding & entropy coding