metadata
license: apache-2.0
tags:
- protocol-reverse-engineering
- network-security
- deep-learning
- pytorch
language:
- en
SAGENET Pre-trained Models
Pre-trained models for SAGENET - a protocol reverse engineering tool for format inference using deep learning.
Model Overview
| Model | Protocol | Size |
|---|---|---|
arp_best.pth |
ARP | 90MB |
dns_best.pth |
DNS | 90MB |
icmp_best.pth |
ICMP | 90MB |
mbtcp_best.pth |
Modbus TCP | 90MB |
nbns_best.pth |
NBNS | 90MB |
ntp_best.pth |
NTP | 90MB |
s7comm_best.pth |
S7comm | 90MB |
smb_best.pth |
SMB | 90MB |
tcp_best.pth |
TCP | 90MB |
udp_best.pth |
UDP | 90MB |
Quick Start
from huggingface_hub import hf_hub_download
# Download a specific model
model_path = hf_hub_download(
repo_id="Prescot12/SAGENET",
filename="checkpoints/tcp_best.pth"
)
Usage
See the main repository for complete usage instructions.
# Test with pre-trained models
python main.py --mode test --protocols tcp udp dns
License
Apache 2.0