|
Box Prediction Rebalancing for Training Single-Stage Object Detectors with Partially Labeled Data
Shafin Haque, R. Austin McEver
NeurIPS Workshop (NeurIPS LMRL), 2022
Paper
 / Poster
We propose Box Prediction Rebalancing for single-stage object detectors to combat the challenge of learning from partially annotated datasets. By randomly removing percentages of negative predictions from our model's loss computation, our model performs better as it learns less from false positives which may be true species of interest without ground truth.
|
|
Glass Imaging
May 2024 - August 2024
Computer Vision & Machine Learning Intern
- Deep learning for mobile image signal processors, working on image restoration tasks such as denoising, demosaicing, enhancement, and super-resolution.
- Lead face restoration project, training efficient SOTA image restoration models with techniques specifically for faces such as GAN-inversion with high quality facial priors.
- Worked directly with RAW sensor captures, enhancing real details while balancing hallucinations.
|
|
Invisible AI
February 2023 - August 2023
Machine Learning Engineering Intern
- Developed few-shot model for semantic segmentation of workpiece objects in automotive manufacturing lines. Trained model with only 15 images per camera site.
- Built light-weight single-shot object detector using features from existing production post estimation model for face blurring.
|
|
UC Santa Barbara, Vision Research Lab
March 2022 - December 2022
Research Intern
- Novel object detection methods for partially labeled datasets, streamlining identification and counting process of marine species in underwater ROV videos.
- Wrote extended abstract accepted to NeurIPS LMLR and contributed to research paper accepted to International Journal of Computer Vision and CVPR Workshop.
|
Projects
More of my projects can be found on my Github or Devpost.
|
|
WildfiresAI
Python, TensorFlow, Flask, HTML/CSS/JS
Code / Devpost / Demo
- Python web app for predicting burn area, put-out time, and location of wildfires with machine learning. Used CNNs, regression, and object detection detection models
- 4th place Congressional App Challenge 2021 & 4x Hackaton Award Winner
|
|
Pitch Prediction
R, Python, RShiny, TensorFlow, Caret
RShiny App / Code / Medium Article
- RShiny app for predicting the next pitch in baseball based on current game scenario. Used regression and neural network based approaches
- Wrote article published in TowardsAI documenting process
|
Website template from here
|
|