Skip to content

This project trains a SSD300 with VGG11 base on the PASCAL VOC 2005 dataset using the PyTorch deep learning framework.

License

Notifications You must be signed in to change notification settings

sovit-123/SSD300-VGG11-on-Pascal-VOC-2005-Data

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

29 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

SSD300-VGG11-on-Pascal-VOC-2005-Data

This project trains a SSD300 with VGG11 base on the PASCAL VOC 2005 dataset using the PyTorch deep learning framework.

Table of Contents

Introduction

This project trains a SSD300 with VGG11 base on the PASCAL VOC 2005 dataset using the PyTorch deep learning framework.

The VOC 2005 dataset mainly contains 4 classes for detection purposes. They are:

  • Bicycle.
  • Car.
  • Motorbike.
  • Person.

This is mainly intended for beginners in deep learning who are trying to get their hands on object detection and do not want to work on very huge datasets.

Feel free to clone, download, change, distribute, and use the code any way you want.

Project Directory Structure

Follow the below given structure while cloning this repo or setting up the project and you should be good to go,

├───VOCdevkit
└───voc_2005
    ├───voc2005_1
    │   ├───Annotations
    │   ├───GTMasks
    │   └───PNGImages
    └───voc2005_2
        ├───Annotations
        ├───GTMasks
        └───PNGImages
├───logs
├───model_checkpoints
├───outputs
└───src
    │   create_data.py
    │   datasets.py
    │   detect.py
    │   detect_vid.py
    │   eval.py
    │   model.py
    │   my_utils.py
    │   train.py
    │   utils.py

If you follow the above structure, then you can just clone and run the files according to the execution commands.

PyTorch Version

  • PyTorch >= 1.4

Running the Code

Execute the following commands in the given order.

Note: For testing purposes, put all the image and video files inside input/test_data that you want to test or detect on.

  • Run the following command only once. This prepare the train and test JSON files inside the input folder.

    • python create_data.py --input1 ../input/VOCdevkit/voc_2005/voc2005_1 --input2 ../input/VOCdevkit/voc_2005/voc2005_2 --save-path ../input
  • To train the SSD model on the dataset.

    • python train.py --input ../input --continue-training yes
    • If you are training for the first time, then give the --continue-training argument as no. This will train the model from scratch.
    • If you have trained the model for some epochs and want to continue training, then give the --continue-training argument as yes.
    • Check the number of iterations to train for inside the train.py file.
  • To run the model and detection on videos.

    • python detect_vid.py --input ../input/test_data/video1.mp4 --resize yes
    • In the above command, if --resize is yes, then the code will resize (make them smaller) the video frames only if the original frames' width is more than 800px.
  • To detect objects in an image.

    • python detect.py --input ../input/test_data/fire_fighter.jpg
  • To run evaluation on validation data.

    • python eval.py

Get Access to Pretrained Model

If you have hardware issues and cannot train the model on your system, then here is the link to a model trained for 156 epochs.

Results

References

About

This project trains a SSD300 with VGG11 base on the PASCAL VOC 2005 dataset using the PyTorch deep learning framework.

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages