Edit model card

JPEGlitch

JPEGlitch is a specialized neural network designed to detect glitches in JPEG images. It combines the architectures of ResNet18 and SPPNet, providing the capability to process images of any size.

Model Architecture

JPEGlitch is structured around an 18-layer Residual Network (ResNet) backbone, enhanced with a Spatial Pyramid Pooling (SPP) layer from SPPNet. This combination ensures that the model can handle input images of arbitrary sizes without the need for resizing or cropping, preserving the integrity and details of the original images. The network consists of approximately 11 million parameters, optimizing the balance between complexity and performance.

Training Dataset

The model is trained on a custom dataset derived from a subset of the ImageNet-1k dataset. This training set includes 500K samples, evenly divided into 250K positive examples (images without glitches) and 250K negative examples (images with glitches introduced through byte flipping).

Optimization and Learning Rate

JPEGlitch employs the AdamW optimizer. The learning rate is set to 1×10−41 \times 10^{-4}.

image/png

Downloads last month
0
Unable to determine this model's library. Check the docs .

Dataset used to train bobqianic/JPEGlitch