From 296907f83147d91e7c3abcefcac99d284e1e705d Mon Sep 17 00:00:00 2001 From: tslgithub Date: 2019年8月26日 18:43:50 +0800 Subject: [PATCH 01/11] Update README.md --- README.md | 3 +++ 1 file changed, 3 insertions(+) diff --git a/README.md b/README.md index a606c67..12ccb08 100644 --- a/README.md +++ b/README.md @@ -55,6 +55,9 @@ My environment is based on * pip3 install opencv-python * pip3 install scikit-learn +# confirm config.py +* choose and change parameter in config.py + # train or test dataset prepare * python3 mk_class_idx.py From b991dcb74e5be6d4d40094ac02d2332507ceaad9 Mon Sep 17 00:00:00 2001 From: tslgithub Date: 2019年8月26日 18:44:28 +0800 Subject: [PATCH 02/11] Update README.md --- README.md | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/README.md b/README.md index 12ccb08..c613ac2 100644 --- a/README.md +++ b/README.md @@ -55,17 +55,17 @@ My environment is based on * pip3 install opencv-python * pip3 install scikit-learn -# confirm config.py +# 1. confirm config.py * choose and change parameter in config.py -# train or test dataset prepare +# 2. train or test dataset prepare * python3 mk_class_idx.py -# train your model +# 3. train your model * train model: python3 train.py modelName * or run " __sh trainAll.sh__ " to train all model -# predict your model +# 4. predict your model * predict model: python3 predict.py model_name classes_name ### Any Questions??? From 4dfa5c667a26327091d2d8541a7570cc24483bda Mon Sep 17 00:00:00 2001 From: tslgithub Date: 2019年8月27日 09:36:46 +0800 Subject: [PATCH 03/11] Update README.md --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index c613ac2..db482ef 100644 --- a/README.md +++ b/README.md @@ -19,7 +19,7 @@ * DenseNet(dismissed this time) -##train or test dataset +## train or test dataset #### classes name contained in folder name From 03d4ea20bf418e39cb38809335b3a4e55de843b7 Mon Sep 17 00:00:00 2001 From: tslgithub Date: 2019年8月27日 09:38:14 +0800 Subject: [PATCH 04/11] Update README.md --- README.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index db482ef..ae32de3 100644 --- a/README.md +++ b/README.md @@ -56,7 +56,7 @@ My environment is based on * pip3 install scikit-learn # 1. confirm config.py -* choose and change parameter in config.py +* choose model and change parameter in config.py # 2. train or test dataset prepare * python3 mk_class_idx.py @@ -66,7 +66,7 @@ My environment is based on * or run " __sh trainAll.sh__ " to train all model # 4. predict your model -* predict model: python3 predict.py model_name classes_name +* predict model: python3 predict.py modelName classesName ### Any Questions??? Author email: mymailwith163@163.com From 5fe08be8491ac7da28744db0ee2ab8d7165d4e25 Mon Sep 17 00:00:00 2001 From: tslgithub Date: 2019年8月27日 09:41:09 +0800 Subject: [PATCH 05/11] Update test_class_idx.txt --- test_class_idx.txt | 1 - 1 file changed, 1 deletion(-) diff --git a/test_class_idx.txt b/test_class_idx.txt index 16c7df6..af7a666 100644 --- a/test_class_idx.txt +++ b/test_class_idx.txt @@ -1,3 +1,2 @@ class_0 class_1 -class_2 \ No newline at end of file From ef0b0b60652b938d3a41fdc4df86edb6504d99db Mon Sep 17 00:00:00 2001 From: tslgithub Date: 2019年8月27日 09:41:46 +0800 Subject: [PATCH 06/11] Update train_class_idx.txt --- train_class_idx.txt | 1 - 1 file changed, 1 deletion(-) diff --git a/train_class_idx.txt b/train_class_idx.txt index 16c7df6..af7a666 100644 --- a/train_class_idx.txt +++ b/train_class_idx.txt @@ -1,3 +1,2 @@ class_0 class_1 -class_2 \ No newline at end of file From 34c3f7ee0f678c9f59eade50642fe1f87b121833 Mon Sep 17 00:00:00 2001 From: tslgithub Date: 2019年8月27日 09:42:34 +0800 Subject: [PATCH 07/11] Update config.py --- config.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/config.py b/config.py index 69477ee..f8ca7f4 100644 --- a/config.py +++ b/config.py @@ -24,9 +24,9 @@ class DefaultConfig(): normal_size = 64 # normal_size = 48 epochs = 30 - batch_size = 8 + batch_size = 2 data_augmentation = False - classes = 49 + classes = 2 channles = 3 # or 3 or 1 lr = 0.00001 From 4c457b0fc5e6437c15d2d30dda3a4b0d45ace952 Mon Sep 17 00:00:00 2001 From: tslgithub Date: 2019年8月29日 09:42:07 +0800 Subject: [PATCH 08/11] Update README.md --- README.md | 1 + 1 file changed, 1 insertion(+) diff --git a/README.md b/README.md index ae32de3..3ce98c3 100644 --- a/README.md +++ b/README.md @@ -64,6 +64,7 @@ My environment is based on # 3. train your model * train model: python3 train.py modelName * or run " __sh trainAll.sh__ " to train all model +* for VGG16,run " __tensorboard --logdir=/home/tsl/PycharmProjects/image_class/checkpoints/VGG16__ " to watch training with tensorboard # 4. predict your model * predict model: python3 predict.py modelName classesName From 5d199a5fb98ac6d694daf1be70c654134972e8a1 Mon Sep 17 00:00:00 2001 From: tslgithub Date: 2019年8月29日 15:45:55 +0800 Subject: [PATCH 09/11] Update README.md --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index 3ce98c3..7edb73d 100644 --- a/README.md +++ b/README.md @@ -64,7 +64,7 @@ My environment is based on # 3. train your model * train model: python3 train.py modelName * or run " __sh trainAll.sh__ " to train all model -* for VGG16,run " __tensorboard --logdir=/home/tsl/PycharmProjects/image_class/checkpoints/VGG16__ " to watch training with tensorboard +* take VGG16 as example, run " __tensorboard --logdir=/home/tsl/PycharmProjects/image_class/checkpoints/VGG16__ " to watch training with tensorboard # 4. predict your model * predict model: python3 predict.py modelName classesName From 7af032b90596f52c2d8f5cfd95e51e0a09432fe4 Mon Sep 17 00:00:00 2001 From: tslgithub Date: 2019年8月31日 15:02:48 +0800 Subject: [PATCH 10/11] Update config.py --- config.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/config.py b/config.py index f8ca7f4..7f6861f 100644 --- a/config.py +++ b/config.py @@ -10,8 +10,12 @@ import sys class DefaultConfig(): - # model_name = 'ResNet34' - model_name = sys.argv[1] + + try: + model_name = sys.argv[1] + except: + print("use default model VGG16") + model_name = 'VGG16' train_data_path = '/media/tsl/DataBackup/TrafficSignalDataset/training/' test_data_path = '/media/tsl/DataBackup/TrafficSignalDataset/testing/' checkpoints = './checkpoints/' From ae35079076ee3c906a6b95f33d4774990c71bd14 Mon Sep 17 00:00:00 2001 From: tslgithub Date: 2019年8月31日 17:06:56 +0800 Subject: [PATCH 11/11] Update README.md --- README.md | 3 +++ 1 file changed, 3 insertions(+) diff --git a/README.md b/README.md index 7edb73d..67f6aa7 100644 --- a/README.md +++ b/README.md @@ -34,6 +34,8 @@ __"training or testing dataset folder is:"__ /path/classes4/*.jpg, + + * Attentions ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! * classes name must be contained in folder name @@ -69,5 +71,6 @@ My environment is based on # 4. predict your model * predict model: python3 predict.py modelName classesName + ### Any Questions??? Author email: mymailwith163@163.com

AltStyle によって変換されたページ (->オリジナル) /