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 &

results matching ""

    No results matching ""