Skip to content

aamna2401/MSDS20018_Project_DLSpring2021

Repository files navigation

MSDS20018_Project_DLSpring2021

Dataset Link

Dataset here

Dependencies

  • numpy
  • torch
  • torchvision
  • Pillow
  • pycocotools
  • matplotlib
  • tensorboardX

Model Training

!python train.py --model_save_dir /mydrive/FasterRCNN/checkpoints

Evaluation Results

Training mAp 0.98
Validation mAP 0.17
Best mAp 0.19

mAp vs epochs

image

Training Loss and Learning Rate

image

Model Testing

!python test.py

Detection Results

image

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published