You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Copy file name to clipboardExpand all lines: RCNN/RCNN.md
+2-1Lines changed: 2 additions & 1 deletion
Display the source diff
Display the rich diff
Original file line number
Diff line number
Diff line change
@@ -4,8 +4,9 @@ The Tensorflow with tflearn implementation of the RCNN model
4
4
**The Enviroment:**
5
5
The file is working with Tensorflow version 0.7.1. Python version 2.7 and scikit-learn dev distribution package. Also need to install the tflearn project from github at https://github.com/tflearn/tflearn. Notice that the tflearn requires Tensorflow version at least 0.7.0. For better compatiblity, you may want to install Tensorflow 0.8.0 or above. Finally, you need the
6
6
selective search project from python. You may get it by pip install selectivesearch.
7
+
7
8
**Training Input:**
8
-
The work trains with the 17 flowers dataset. The data are provided together with the project.
9
+
The work trains with the 17 flowers dataset. The data can be obtained here: https://github.com/ck196/tensorflow-alexnet. Download the project and use the 17flowers.tar.gz file's data as the training data.
9
10
10
11
**The Code:**
11
12
There are four files in the code. The train_alexnet.py uses the 17flowers image folder with the train_list.txt file to perform the pre-training of the Alexnet. This is the file that need to be run first. After it generates the model_save.model file, you can run the fine_tune_RCNN.py file, which fine-tunes the model with the 2flowers dataset as well as the two txt files in
0 commit comments