Malware Traffic Classification Using Convolutional Neural Network for Representation Learning代码复现的问题

论文下载：https://ieeexplore.ieee.org/document/7899588

代码下载：https://github.com/echowei/DeepTraffic

数据处理工具包使用方法：https://github.com/yungshenglu/USTC-TK2016

作者博士毕业论文收录于知网：基于深度学习的网络流量分类及异常检测方法研究
————————————————
原文链接：https://blog.csdn.net/caiguanhong/article/details/109695999

在进行代码复现前首先要对数据集进行预处理，预处理部分在其他文章有很多详细介绍，在此不做赘述，本文只简单介绍在复现代码traffic_cnn.py时自己遇到的一些问题。

作者的代码是在tensorflow1.x下写的，我的版本时tf2.9，所以会出现的问题较多。

1.input_data不存在

将import input_data换成import tensorflow.examples.tutorials.mnist.input_data ，若没有tenfowflow.examples，则先在pycharm中新建一个文件

import tensorflow as tf

print(tf.__file__)

查看你的路径。

我的路径是：D:\Anaconda3\envs\TF2.4.0\lib\site-packages\tensorflow，在路径中创建examples文件夹，并将下载好的tutorials包复制进去即可。（tutorials包在网上有很多资源，自行百度下载）

2.报错DATA_DIR = sys.argv[1] IndexError: list index out of range，

这表明没有传入参数，在传入参数DATA_DIR、CLASS_NUM、TRAIN_ROUND时，可以直接写成

DATA_DIR="D:\\code\\tensorflow\\deeptrafic\\DeepTrafficmaster\\1.malware_traffic_classification\\4.TrainAndTest\\5_Mnist"
CLASS_NUM = 10
TRAIN_ROUND = 400

这种形式，也可以在pycharm中自己设置，点击“Run”菜单下的"Edit Configurations..."命令，在parameter中输入三个参数的值，每个参数输入完后，需要有一个空格。

3.报错 CLASS_NUM = int(sys.argv[2]) ValueError: invalid literal for int() with base 10: '='

将上面的代码

DATA_DIR = sys.argv[1]

CLASS_NUM = int（sys.argv[2]）

TRAIN_ROUND = int （sys.argv[3]）

中的int去掉，变为：

DATA_DIR = sys.argv[1]

CLASS_NUM = sys.argv[2]

TRAIN_ROUND = sys.argv[3]

5.报错 sess = tf.InteractiveSession() AttributeError: module 'tensorflow' has no attribute 'InteractiveSession'

将代码改为

sess = tf.compat.v1.InteractiveSession()

6.palceholder报错

Tf2.x中placeholder应该使用tf.compat.v1.placeholder，并在开头import tensorflow as tf后面加上tf.compat.v1.disable_eager_execution()

还有一些问题基本上都是tf1.x中的函数转换为tf2.x的问题，百度都可以找到。

Malware Traffic Classification Using Convolutional Neural Network for Representation Learning代码复现的问题相关推荐

【农业害虫识别论文一】Crop pest classification based on deep convolutional neural network and transfer learning
本论文发布于2019年,主要基于三个数据集进行农业害虫的识别.文章不是简单的翻译,而是类似自己的读书笔记,基本记录了下,做后续回顾所用.望各位看官不要吐槽哈哈! 论文题目 Crop pest cla ...
PRN（20200816）：A Hierarchical Deep Convolutional Neural Network for Incremental Learning [Tree-CNN]
Roy D , Panda P , Roy K . Tree-CNN: A Hierarchical Deep Convolutional Neural Network for Incremental ...
Neural Networks for Multi-Instance Learning代码复现
Neural Networks for Multi-Instance Learning论文阅读笔记多示例数据的数据集结构与单实例的数据不同,一个样本由一个包表示,一个包中有多个实例,标签被分配给包而 ...
卷积神经网络CNN（Convolutional Neural Network）原理与代码实现 Le-Net5
图像识别经典数据集: 图像识别是人工智能的一个重要的领域.其他常用的图像识别数据集: CIFAR: http://www.cs.toronto.edu/~kriz/cifar.html CIFAR数 ...
VGG Convolutional Neural Network Practical
系列文章目录 VGG Image Classification Practical Convolutional Neural Network Practical 系列文章目录前言一.准备工作 1. ...
读后感-论文Patch-based Convolutional Neural Network for Whole Slide Tissue Image Classification
读后感背景工作中要对高分辨率图像(至少1024 \(\times\) 1024)进行分类,而且ground-truth分类的标准一般都是取决于像素级别的精细local feature(e.g. te ...
Paper翻译：《A Novel Convolutional Neural Network Based Model for Recognition and Classification of App》
论文名称:<A Novel Convolutional Neural Network Based Model for Recognition and Classification of Appl ...
HD-CNN: HIERARCHICAL DEEP CONVOLUTIONAL NEURAL NETWORK FOR IMAGE CLASSIFICATION(泛读)
一.文献名字和作者 HD-CNN: HIERARCHICAL DEEP CONVOLUTIONAL NEURAL NETWORK FOR IMAGE CLASSIFICATION, 2014 二 ...
《Improved Crowd Counting Method Based onScale-Adaptive Convolutional Neural Network》论文笔记
<Improved Crowd Counting Method Based onScale-Adaptive Convolutional Neural Network>论文笔记论文地址 ...

Malware Traffic Classification Using Convolutional Neural Network for Representation Learning代码复现的问题

Malware Traffic Classification Using Convolutional Neural Network for Representation Learning代码复现的问题相关推荐

最新文章

热门文章