Skip to content

Commit 917deef

Browse files
authored
Update RCNN.md
1 parent 4ea4310 commit 917deef

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

RCNN/RCNN.md

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,8 +4,9 @@ The Tensorflow with tflearn implementation of the RCNN model
44
**The Enviroment:**
55
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
66
selective search project from python. You may get it by pip install selectivesearch.
7+
78
**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.
910

1011
**The Code:**
1112
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

Comments
 (0)