- Colab
- Keras
- Tensorflow
- Pandas
In first step, separate images into train,valid and test directories. The dataset was provided in Pascal voc but was converted to custom format
of filepath,x1,y1,x2,y2,class_name
. Training requires annotation files in .txt
format. For step by step preprocessing please refer to
Preprocessing/FRCNN_preprocessing.ipynb
Example dataset provided in Dataset
folder
Refer to frcnn_train_vgg.ipynb
Refer to frcnn_test_vgg.ipynb
.
Note: training config is used for testing the Model.
Model | Validation mAP | Test mAP |
---|---|---|
Faster-RCNN | 0.515 | 0.508 |
Pretrained weights can be from keras/application
Image Size = 512
Number of RoIs at once = 4
Data Augumentation: Horizontal flips, Vertical Flips, rotation 90 degree.
RPN optimizer: Adam, Learning rate =1e-5
Classification optimizer: Adam, Learning rate =1e-5
Overall optimizer: SGD
Loss function:Mean Absolute Error (MAE)