Faster-RCNN [keras]
https://github.com/yhenon/keras-frcnn
非常好用的一个版本,代码简洁,自行训练也行方便,不需要预先下载VGG,ResNet等模型参数,keras有自带的。
Problem
long time to train > 7 days
do not support tensorboard
error for load trained model
Traceback (most recent call last): File "test_frcnn.py", line 36, in <module> C = pickle.load(f_in) File "/usr/lib/python2.7/pickle.py", line 1384, in load return Unpickler(file).load() File "/usr/lib/python2.7/pickle.py", line 864, in load dispatch[key](self) KeyError: 'H'
Train
nohup python2 train_frcnn.py -o simple -p iscas_3c.txt --output_weight_path ./iscas_3c_keras_frcnn.h5 &