Torchvision Faster RCNN Fine Tuner
Project description
Pytorch Faster RCNN
Faster RCNN Fine-Tune Implementation in Pytorch.
How to use ?
- git clone the repo
git clone https://github.com/oke-aditya/pytorch_fasterrcnn.git
- Install PyTorch and torchvision for your system.
Simply edit the config file to set your hyper parameters.
- Keep the training and validation csv file as follows
NOTE
Do not use target as 0 class. It is reserved as background.
image_id xtl ytl xbr ybr target
1 xmin ymin xmax ymax 1
1 xmin ymin xmax ymax 2
2 xmin ymin xmax ymax 3
-
Simply edit the config file to set your hyper parameters
-
Run the train.py file
Features: -
- It works for multiple class object detection.
Backbones Supported: -
-
Note that backbones are pretrained on imagenet.
-
Following backbones are supported
- vgg11, vgg13, vgg16, vgg19
- resnet18, resnet34, resnet50, resnet101, resnet152
- renext101
- mobilenet_v2
Sample Outputs
Helmet Detector
Mask Detector
If you like the implemenation or have taken an inspiration do give a star :-)
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Close
Hashes for pytorch_fasterrcnn-0.2.0rc2.tar.gz
Algorithm | Hash digest | |
---|---|---|
SHA256 | c773af6be3a76a415be4a0c2d05f6ad510a6b9f8fad175f1333077c176f67217 |
|
MD5 | a13417943a883e8c887f593177c4d529 |
|
BLAKE2b-256 | f741f74665f6ffd65331ce109f24499ec7c03965a04ddf8f89ba930499064626 |
Close
Hashes for pytorch_fasterrcnn-0.2.0rc2-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 829152971369e3dcc542afd8e1f0b783f3cfd461e0fab90701f39ff481724fe4 |
|
MD5 | 129f33054ce63b6fbd00e41e6263bdd8 |
|
BLAKE2b-256 | be695a1fc51628dceb1c82b8510d70ac623b337e5e38dc94b99dcad2761e38ba |