Implement a single object detector with PyTorch, and train on Tiny-VID dataset. For each classes, use 150 samples for training and the rest for testing.
- Watch all parts till Object Detection of CS231n.
- Paper reading: Faster R-CNN, [Optional] YOLO
- PyTorch Tutorials: https://pytorch.org/tutorials. Only Introduction to PyTorch and Learning PyTorch parts are needed.
- Design a detector, and train & evaluate on Tiny-VID dataset.
Complete
TvidDataset
,Detector
,compute_iou
and the rest inmain.py
. The accuracy should be at least 70%. - Visualize detection results.
- Expected completion time: 2~3 days.