/train-faster-rcnn

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

Primary LanguagePython

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