Skip to content

sebasegovia01/CNNYoloV3

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

CNNYoloV3 WHW

CNN for analytical traffic

Requirements:

  • TensorFLow 2.0
  • cudatoolkit 10.0
  • cudnn 7.6
  • opencv 4.2

If you use Anaconda these two lines will solve the Requirements.

For GPU users:

 conda install -c conda-forge tensorflow-gpu=2.0
 
 conda install -c conda-forge opencv

For CPU users:

 conda install -c conda-forge tensorflow=2.0
 
 conda install -c conda-forge opencv

Execution Steps:

  1. Download the "yolov3.weights", from: https://pjreddie.com/media/files/yolov3.weights to the weights folder.

  2. Execute the convert_weights.py:

    python convert_weights.py

    Before testing, make sure that the weights have been converted to TF2 format. The converted weights file is saved in the weights folder.

#Testing

  1. For the image:

    python image.py

  2. For the video/cam:

    python video.py

About

CNN WHW for analytical traffic

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages