- Developed a custom object detection and instance segmentation model using Mask R-CNN and Faster R-CNN architectures, built on top of pre-trained CNN backbones such as ResNet-50 and MobileNetV2.
- Used the PyTorch deep learning framework to train and evaluate the models on a custom dataset of images and annotations, achieving high accuracy and performance metrics.
- Applied techniques such as transfer learning, fine-tuning, and data augmentation to improve the performance of the models on the target task, and optimized the training process using various hyperparameters and loss functions.
- Visualized the predictions of the models using various tools and libraries such as Matplotlib, NumPy, and OpenCV, and provided insights and feedback to improve the models based on the observed results.
-
Notifications
You must be signed in to change notification settings - Fork 0
rr2203/torchvision-object-detection-finetuning
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
Repository files navigation
About
Modifying pre-trained torchvision models.
Topics
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published