项目作者: Nebula4869
项目描述 :
Real-time object detection using COCO-pretrained EfficientDet under Pytorch and TensorFlow
高级语言: Python
项目地址: git://github.com/Nebula4869/real-time_object_detection_EfficientDet.git
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
- Download checkpoint and convert to .pb model files from this repo.
- Download Pytorch .pth model files from this repo
- Run “demo_pytorch.py” to detect with Pytorch model.
- Run “demo_tf.py” to detect with TensorFlow frozen model.