Skip to content

This YoloV5 based model is fit to detect people and different types of land vehicles, and displaying their density on a fitted map, according to their coordinates and detected labels.

License

lironbdolah/streetdensityai

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Street Density ai

Licence Issues last commit

This YoloV5 based model is fit to detect people and different types of land vehicles,
and displaying their density on a fitted map, according to their coordinates and detected labels.

Features:

  • Multiple Objects Detection
  • Trained on 3,000 street view Images
  • Exports Fitted and adjustable Maps
  • Calculates a density score according to image detected labels

Requierments:

Usage:

Object Detection:

identifies people and land vehicles in your images:

python src/yolov5/detect.py --source  <path to images folder> --project <output path>
--name <output folder name> --save-txt --conf 0.3

running this action will save your images with the anchor boxes around objects that the model found:

(if you don't want to save the labeled images, just add --nosave to the command above) in addition, it will save the detected object labels for each image.

Plotting a fitted map:

display the density on a fitted map (requires a .csv file)

python src/steetdensityai.py --labels <labels path that were created after the images detection>
--coordinates <path-to-csv/file.csv>  --images <path to images folder>
--img-per-cord 1 --output <output path>

notes

  • csv requires 2 columns to display the coordinates named: "longitude" and "latitude"
  • the code asumes that the coordinates are sorted by the image's name.
  • If you have multiple images per coordinate (for example if you have a 360 view, divided to 4 images), you can set the number of images per coordinate with : --img-per-cord <integer of images per coordinate >

Simple Example:

# detect objects: 
python src/yolov5/detect.py --source example/images --project example/images --name detected_images --save-txt --conf 0.4


# creates a label folder in example/images/detected_images named "labels"
# saves the images with the newly found objects anchor, and each image labels 


#plot desnity map
python src/steetdensityai.py --labels example/images/detected_images/labels --coordinates example/coordinates.csv  --images example/images --img-per-cord 4 --output example/images

 # will save the map.html file to example/images

About

This YoloV5 based model is fit to detect people and different types of land vehicles, and displaying their density on a fitted map, according to their coordinates and detected labels.

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages