Skip to content

pranjalsrajput/Video_Image_Annotator_YoloV3

Repository files navigation

Video_Image_Annotator_YoloV3

For bounding box and labels based annotation of videos using YoloV3

  1. Clone the repository and install requirements

    cd PyTorch-YOLOv3/

    sudo pip3 install -r requirements.txt

  2. Download pretrained weights

    cd weights/

    bash download_weights.sh

  3. Put your data in data/samples/ folder

  4. Run python3 detect.py --image_folder data/samples/

  5. The system will take each image/frame and will ask to either annotate(Press Continue) or not(Press Skip)?

drawing

Figure 1: Window 1

  1. If Skip is pressed, then next image/frame will be taken. If Continue is pressed then you will be asked to put the label value (in our case it's bibId of the runners). You can Enter a value or Quit if don't want to put any label for that object (in this case runner).

drawing

Figure 2: Window 2

  1. In the end, the annotateed images will be saved in the output folder and a .csv file will be saved containing the bounding box and label values.

References

Above repo is retrieved from

Credit

YOLOv3: An Incremental Improvement Joseph Redmon, Ali Farhadi

Abstract

We present some updates to YOLO! We made a bunch of little design changes to make it better. We also trained this new network that’s pretty swell. It’s a little bigger than last time but more accurate. It’s still fast though, don’t worry. At 320 × 320 YOLOv3 runs in 22 ms at 28.2 mAP, as accurate as SSD but three times faster. When we look at the old .5 IOU mAP detection metric YOLOv3 is quite good. It achieves 57.9 AP50 in 51 ms on a Titan X, compared to 57.5 AP50 in 198 ms by RetinaNet, similar performance but 3.8× faster. As always, all the code is online at https://pjreddie.com/yolo/.

@article{yolov3, title={YOLOv3: An Incremental Improvement}, author={Redmon, Joseph and Farhadi, Ali}, journal = {arXiv}, year={2018} }

About

For bounding box and labels based annotation of videos using YoloV3

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published