keras-retinanet环境搭建
retinanet的keras版本,github地址:fizyr/keras-retinanet
搭建过程
- 建立并激活conda环境
conda create -n retinanet python=3.6
conda activate retinanet
- 安装numpy,根据cuda版本,安装tensorflow及keras,参考tensorflow各个版本的CUDA以及Cudnn版本对应关系,这里选用了tensorflow-gpu==1.12.0
pip install numpy --user
pip install --ignore-installed --upgrade tensorflow-gpu==1.12.0
- 路径根据自己情况选择
cd git
git clone https://github.com/fizyr/keras-retinanet.git
cd keras-retinanet
pip install . --user
python setup.py build_ext --inplace(可选)
- 要执行example里的demo,还需在此conda 环境下:
pip install jupyter
pip install imutils