Skip to content

Train Faster R-CNN on custom dataset (including training plots)

Notifications You must be signed in to change notification settings

enzeberg/train-faster-rcnn

Repository files navigation

Train

python train.py --config data_configs/fishes.yaml --epochs 70 --batch-size 10 --results-folder fishes --use-train-aug --no-mosaic

Training Plots

mAP train_loss_epoch train_loss_iter Box regression loss Classification loss Object loss RPN bounding box loss

Inference

​python inference.py --weights runs/train/fishes/best_model.pth --input D:/CV/datasets/fishes_voc/test

mAP mAP mAP mAP mAP

About

Train Faster R-CNN on custom dataset (including training plots)

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages