Updata Transfer Learning

Author: Kedreamix

TransferLearning/README.md

@@ -14,9 +14,26 @@ python train.py --data  './cats_and_dogs_filtered/'
 		--classes 2 # set classes 
     		--epochs 10 # set epochs
         	--lr  0.02 # set learning rate
-            	--optimizer Adam # 优化器 如Adam,AdamW,SGD
+            	--optimizer 'Adam' # 优化器 如Adam,AdamW,SGD
                 --batch-size 64 # batch size
                 --verbose # 可视化
                 --logs # 日志，每次都保存模型
 ```
 
+## 使用自己的数据集
+
+这里要注意，对于自己的数据集，需要按照一定的排列格式，可以按照你的数据集排列方式，最后在运行的时候改成python train.py --data './data'即可
+
+```txt
+data
+	----train
+    		----dogs
+         	----cats
+         	----...
+    ----validation
+    		----dogs
+         	----cats
+         	----...
+   	----test(放任意图片，可以在命令行设置，没有默认0)
+```
+

TransferLearning/train.py

@@ -93,7 +93,7 @@ def main(opt):
     parser.add_argument('--save-path',type=str,default='model.pth',help='save model')
     parser.add_argument('--verbose',action='store_true',help='可视化训练结果')
     parser.add_argument('--logs',action='store_true',help='每次迭代后都保存模型')
-    parser.add_argument('--test-size',type=int,default=1000,help='the number for picture in data/test')
+    parser.add_argument('--test-size',type=int,default=0,help='the number for picture in data/test')
     # parser.add_argument('--device', default='', help='cuda device, i.e. 0 or 0,1,2,3 or cpu') # 可以自动选择GPU，但是这里我就让程序自动判断
 
     opt = parser.parse_args()