windows环境下使用EfficientDet(一)

1、安装TensorFlow 2.1.0、cuda10.1、cudnn7.6.5

  参见:https://www.cnblogs.com/zhengbiqing/p/12554992.html

2、安装pycocotools

  参见:https://www.cnblogs.com/zhengbiqing/p/12552775.html

3、下载预训练模型:

  https://storage.googleapis.com/cloud-tpu-checkpoints/efficientdet/coco/efficientdet-d6.tar.gz

4、识别一张图片:

  打开命令行,执行:

  SET PYTHONPATH=.

  python model_inspect.py --runmode=infer --model_name=efficientdet-d6 --ckpt_path=E:\efficientdet-d6\efficientdet-d6\ --input_image=E:\efficientDect\bird.jpg --output_image_dir=E:\

识别结果:

 

posted @ 2020-05-03 11:46  碧水青山  阅读(1003)  评论(0编辑  收藏  举报