Model Summary
ForensicHub supports all IMDLBenCo and DeepfakeBench models, and newly implements models for AIGC-generated image detection and document tampering detection, as summarized in the table below. The registered names are the names users need to use when employing ForensicHub:
| Domain | Model | Paper Link | Code Link | Registered Name | Code Path |
|---|---|---|---|---|---|
| Backbone | ConvNeXt | Paper | GitHub | ConvNextSmall | ForensicHub/common/backbones/convnext.py |
| Backbone | DenseNet | Paper | GitHub | DenseNet121 | ForensicHub/common/backbones/densenet.py |
| Backbone | EfficientNet | Paper | GitHub | EfficientB4 | ForensicHub/common/backbones/efficientnet.py |
| Backbone | MobileNet | Paper | GitHub | MobileNetV3 | ForensicHub/common/backbones/mobilenet.py |
| Backbone | ResNet | Paper | GitHub | Resnet50 | ForensicHub/common/backbones/resnet.py |
| Backbone | SegFormer | Paper | GitHub | Segformerb3 | ForensicHub/common/backbones/segformer.py |
| Backbone | Swin Transformer | Paper | GitHub | SwinSmall | ForensicHub/common/backbones/swintransformer.py |
| Backbone | UNet | Paper | GitHub | UNetBase | ForensicHub/common/backbones/unet.py |
| Backbone | ViT | Paper | GitHub | ViTBase | ForensicHub/common/backbones/vit.py |
| Backbone | Xception | Paper | GitHub | Xception | ForensicHub/common/backbones/xception.py |
| AIGC | Dire | Paper | GitHub | Dire | ForensicHub/tasks/aigc/models/dire/dire.py |
| AIGC | DualNet | Paper | GitHub | DualNet | ForensicHub/tasks/aigc/models/dualnet/DualNet.py |
| AIGC | HiFiNet | Paper | GitHub | HiFi_Net | ForensicHub/tasks/aigc/models/hifi_net/HiFi_Net.py |
| AIGC | Synthbuster | Paper | - | Synthbuster | ForensicHub/tasks/aigc/models/synthbuster/Synthbuster.py |
| AIGC | UnivFD | Paper | Github | UnivFD | ForensicHub/tasks/aigc/models/univfd/UnivFD.py |
| Document | CAFTB_Net | Paper | - | CAFTB_Net | ForensicHub/tasks/document/models/caftb_net/caftb_net.py |
| Document | DTD | Paper | Github | DTD | ForensicHub/tasks/document/models/dtd/dtd.py |
| Document | FFDN | Paper | - | FFDN | ForensicHub/tasks/document/models/ffdn/ffdn.py |
| Document | PSNet | Paper | - | PSNet | ForensicHub/tasks/document/models/psnet/psnet.py |
| Document | Tifdm | Paper | - | Tifdm | ForensicHub/tasks/document/models/tifdm/tifdm.py |