File size: 1,187 Bytes
3475a6d
 
5429965
 
 
 
 
 
 
3475a6d
5429965
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
---

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

```python

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](https://github.com/infinityay/SAGENET) for complete usage instructions.

```bash

# Test with pre-trained models

python main.py --mode test --protocols tcp udp dns

```

## License

Apache 2.0