Update README.md

README.md

@@ -6,6 +6,10 @@ Segment_SNN主要实现了一个利用VGG16和FPN作为backbone，采用encoder-
 + 第二阶段我们对FPN网络也进行了转换，并且将其与上一步转换过的VGG16网络进行拼接得到我们的语义分割模型Segment_SNN
 
 文件内容：
++ 数据文件夹结构
+  + datasets
+    + coco
+      + val2017
 + basic.py文件包含了所有的神经元以及基础模块的定义
 + dataset.py文件定义了数据的预处理方式（包含快速眼动生成序列帧以及数据增强）
 + fpn.py文件包含转换后的脉冲FPN网络
@@ -17,9 +21,7 @@ Segment_SNN主要实现了一个利用VGG16和FPN作为backbone，采用encoder-
 #### 数据集准备
 + 使用coco数据训练
 + 数据使用快速眼动法生成dvs帧（共9帧）作为模型输入，方法参考`Lin Y, Ding W, Qiang S, et al. Es-imagenet: A million event-stream classification dataset for spiking neural networks[J]. Frontiers in neuroscience, 2021, 15: 1546.`
-+ 如果使用个人数据集，确保包含如下目录和文件
-  + xxx
-  + xxx
+
 #### 模型训练
 ```python
 python train.py --batch_size 8 --step 8 --learning_rate 0.01 --num_epochs 100 -output_size (480, 480)
@@ -40,6 +42,7 @@ python predict.py --image_path './test/img' --step 8 --output_size (480, 480) --
 #### 第一阶段（分类）
 + 我们在nminst数据集上训练17个epoch后的结果
 <img width="503" alt="94d9608c9238ed6bfae8465e9da21d9" src="https://github.com/yahuiwei123/segment_snn/assets/84215971/99bc2e72-d151-4a2b-bdce-2b81c9982185">
++ 训练的Kaggle Notebook（包括FCN和SNN）：https://www.kaggle.com/code/littleweakweak/test-pycocotools
 
 #### 第二阶段（分割）
 ### 成员分工