Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
yahuiwei123 authored Dec 20, 2023
1 parent b8f4f2c commit 36b7283
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -9,14 +9,14 @@ Segment_SNN主要实现了一个利用VGG16和FPN作为backbone,采用encoder-
+ xxx
+ xxx
##### 模型训练
``
```python
python train.py --batch_size 8 --step 8 --learning_rate 0.01 --num_epochs 100 -output_size (480, 480)
``
```

##### 模型预测
``
```python
python predict.py --image_path './test/img' --step 8 --output_size (128, 128) --output_dir './test/out'
``
```
#### 模型细节
+ 对模型中的如下模块进行了替换
+ Conv2D $\rightarrow$ LayerWiseConvModule
Expand Down

0 comments on commit 36b7283

Please sign in to comment.