项目作者: Nebula4869

项目描述 :
Real-time object detection using COCO-pretrained EfficientDet under Pytorch and TensorFlow
高级语言: Python
项目地址: git://github.com/Nebula4869/real-time_object_detection_EfficientDet.git
创建时间: 2020-06-09T15:53:49Z
项目社区:https://github.com/Nebula4869/real-time_object_detection_EfficientDet

开源协议:MIT License

下载


real-time-object-detection-EfficientDet

Real-time object detection using COCO-pretrained EfficientDet under Pytorch and TensorFlow

Environment

  • python==3.6.5
  • torch==1.5.0
  • torchvision==0.6.0
  • tensorflow==1.15.0
  • opencv-python

Getting Started

  1. Download checkpoint and convert to .pb model files from this repo.
  2. Download Pytorch .pth model files from this repo
  3. Run “demo_pytorch.py” to detect with Pytorch model.
  4. Run “demo_tf.py” to detect with TensorFlow frozen model.