Nebula4869 / real-time-object-detection-EfficientDet Public

Notifications You must be signed in to change notification settings
Fork 1
Star 7

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

7 stars 1 fork Branches Tags Activity

Notifications

Name		Name	Last commit message	Last commit date
Latest commit History 2 Commits
efficientdet		efficientdet
efficientnet		efficientnet
models		models
.gitignore		.gitignore
LICENSE		LICENSE
README.md		README.md
coco.names		coco.names
demo_pytorch.py		demo_pytorch.py
demo_tf.py		demo_tf.py

Repository files navigation

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.

About

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

python tensorflow pytorch object-detection efficientdet

Report repository

Releases

No releases published

Packages

No packages published

Languages

Python 100.0%