Details in blog post: https://blog.munhou.com/2020/04/05/Pytorch-Implementation-of-Deep-Packet-A-Novel-Approach-For-Encrypted-Tra%EF%AC%83c-Classi%EF%AC%81cation-Using-Deep-Learning/
chat
, file_transfer
, voip
, streaming
and vpn_voip
petastorm
, use huggingface's datasets
instead for data loaderconda env create -f env_mac.yaml
conda env create -f env_linux_cpu.yaml
conda env create -f env_linux_cuda102.yaml
conda env create -f env_linux_cuda113.yaml
python preprocessing.py -s /path/to/CompletePcap/ -t processed_data
python create_train_test_set.py -s processed_data -t train_test_data
Application Classification
For CNN model
python train_cnn.py -d train_test_data/application_classification/train.parquet -m model/application_classification.cnn.model -t app
For Resnet model
python train_resnet.py -d train_test_data/application_classification/train.parquet -m model/application_classification.cnn.model -t app
Traffic Classification
For CNN model
python train_cnn.py -d train_test_data/traffic_classification/train.parquet -m model/traffic_classification.cnn.model -t traffic
For Resnet model
python train_resnet.py -d train_test_data/traffic_classification/train.parquet -m model/traffic_classification.cnn.model -t traffic
Download the pre-trained CNN models here.
Code ran on AWS c5.4xlarge
7:01:32 elapsed
Code ran on AWS c5.4xlarge
2:55:46 elapsed
Code ran on AWS g5.xlarge
24:41 elapsed
Code ran on AWS g5.xlarge
7:55 elapsed