Commit 3575751

author

xyliao

committed

fix readme

1 parent 3ffe6aa commit 3575751Copy full SHA for 3575751

File tree

+66

-16

lines changed

+66

-16

lines changed

Lines changed: 1 addition & 1 deletion

Original file line number	Diff line number	Diff line change
`@@ -75,7 +75,7 @@ Learn Deep Learning with PyTorch`
`75`	`75`	`### part2: 深度学习的应用`
`76`	`76`	`- Chapter 9: 计算机视觉`
`77`	`77`	`- [Fine-tuning: 通过微调进行迁移学习](https://github.com/SherlockLiao/code-of-learn-deep-learning-with-pytorch/blob/master/chapter9_Computer-Vision/fine_tune/fine-tune.ipynb)`
`78`		`- - [语义分割: 通过 FCN 实现像素级别的分类](https://github.com/SherlockLiao/code-of-learn-deep-learning-with-pytorch/blob/master/chapter9_Computer-Vision/segmentation/fcn.ipynb)`
	`78`	`+ - [语义分割: 通过 FCN 实现像素级别的分类](https://github.com/SherlockLiao/code-of-learn-deep-learning-with-pytorch/tree/master/chapter9_Computer-Vision/segmentation)`
`79`	`79`	`- Pixel to Pixel 生成对抗网络`
`80`	`80`	`- Neural Transfer: 通过卷积网络实现风格迁移`
`81`	`81`	`- Deep Dream: 探索卷积网络眼中的世界`

Lines changed: 52 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,52 @@`
	`1`	`+## Fine Tune 教程`
	`2`	`+`
	`3`	`+### Requirements`
	`4`	`+`
	`5`	`+[PyTorch 0.3](http://pytorch.org/)`
	`6`	`+`
	`7`	`+[MxTorch](https://github.com/SherlockLiao/mxtorch)`
	`8`	`+`
	`9`	`+[tensorboardX](https://github.com/lanpa/tensorboard-pytorch)`
	`10`	`+`
	`11`	`+按照 pytorch 官网安装 pytorch,将 mxtorch 下载下来,放到根目录,安装 tensorboardX 实现 tensorboard 可视化`
	`12`	`+`
	`13`	+```bash
	`14`	`+\fine_tune`
	`15`	`+ \mxtorch`
	`16`	`+ \hymenoptera_data`
	`17`	`+ \train`
	`18`	`+ \val`
	`19`	`+ \checkpoints`
	`20`	`+ config.py`
	`21`	`+ main.py`
	`22`	`+ get_data.sh`
	`23`	+```
	`24`	`+`
	`25`	`+`
	`26`	`+`
	`27`	`+### 下载数据`
	`28`	`+`
	`29`	`+打开终端,运行 bash 脚本来获取数据`
	`30`	`+`
	`31`	+```bash
	`32`	`+bash get_data.sh`
	`33`	+```
	`34`	`+`
	`35`	`+`
	`36`	`+`
	`37`	`+### 训练模型`
	`38`	`+`
	`39`	`+所有的配置文件都放在 config.py 里面,通过下面的代码来训练模型`
	`40`	`+`
	`41`	+```bash
	`42`	`+python main.py train`
	`43`	+```
	`44`	`+`
	`45`	`+也可以在终端修改配置,比如改变 epochs 和 batch_size`
	`46`	`+`
	`47`	+```bash
	`48`	`+python main.py train \`
	`49`	`+ --max_epochs=100 \`
	`50`	`+ --batch_size=16`
	`51`	+```
	`52`	`+`

Lines changed: 4 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,4 @@`
	`1`	`+# Download data.`
	`2`	`+wget https://download.pytorch.org/tutorial/hymenoptera_data.zip`
	`3`	`+`
	`4`	`+unzip hymenoptera_data.zip`

Lines changed: 4 additions & 12 deletions

Original file line number	Diff line number	Diff line change
`@@ -6,9 +6,9 @@`
`6`	`6`
`7`	`7`	`[MxTorch](https://github.com/SherlockLiao/mxtorch)`
`8`	`8`
`9`		`-[tensorboardX](https://github.com/lanpa/tensorboard-pytorch) (optional)`
	`9`	`+[tensorboardX](https://github.com/lanpa/tensorboard-pytorch)`
`10`	`10`
`11`		`-按照 pytorch 官网安装 pytorch,将 mxtorch 下载下来,放到根目录,同时可以安装 tensorboardX 实现 tensorboard 可视化`
	`11`	`+按照 pytorch 官网安装 pytorch,将 mxtorch 下载下来,放到根目录,安装 tensorboardX 实现 tensorboard 可视化`
`12`	`12`
`13`	`13`	```bash
`14`	`14`	`\segmentation`
`@@ -37,26 +37,18 @@ bash get_data.sh`
`37`	`37`
`38`	`38`	`所有的配置文件都放在 config.py 里面,通过下面的代码来训练模型`
`39`	`39`
`40`		-```python
	`40`	+```bash
`41`	`41`	`python main.py train`
`42`	`42`	```
`43`	`43`
`44`	`44`	`也可以在终端修改配置,比如改变 epochs 和 batch_size`
`45`	`45`
`46`		-```python
	`46`	+```bash
`47`	`47`	`python main.py train \`
`48`	`48`	`--max_epochs=100 \`
`49`	`49`	`--batch_size=16`
`50`	`50`	```
`51`	`51`
`52`		`-#### 可选`
`53`		`-`
`54`		`-如果安装了 tensorboardX,则可以使用 tensorboard 可视化,通过`
`55`		`-`
`56`		-```bash
`57`		`-python main.py train --vis_dir='./vis'`
`58`		-```
`59`		`-`
`60`	`52`
`61`	`53`
`62`	`54`	`### 训练效果`

Lines changed: 5 additions & 3 deletions

Original file line number	Diff line number	Diff line change
`@@ -22,7 +22,7 @@ class DefaultConfig(object):`
`22`	`22`	`save_best = True # If save best test metric model.`
`23`	`23`
`24`	`24`	`# Visualization parameters.`
`25`		`- # vis_dir = './vis/'`
	`25`	`+ vis_dir = './vis/'`
`26`	`26`	`plot_freq = 30 # plot in tensorboard every N iterations`
`27`	`27`
`28`	`28`	`# Model hyperparameters.`
`@@ -47,8 +47,10 @@ def _parse(self, kwargs):`
`47`	`47`	`print('============end===============')`
`48`	`48`
`49`	`49`	`def _state_dict(self):`
`50`		`- return {k: getattr(self, k) for k, _ in DefaultConfig.__dict__.items()`
`51`		`- if not k.startswith('_')}`
	`50`	`+ return {`
	`51`	`+ k: getattr(self, k)`
	`52`	`+ for k, _ in DefaultConfig.__dict__.items() if not k.startswith('_')`
	`53`	`+ }`
`52`	`54`
`53`	`55`
`54`	`56`	`opt = DefaultConfig()`

Comments

(0)