Stereo Matching by Training a Convolutional Neural Network to Compare Image Patches
Role in this project:
ML Engineer Contributions:37 commits, 24 pushes, 1 branch in 1 year 11 months
Contributions summary:Jure primarily contributed to the core functionality of the `mc-cnn` repository, which focuses on stereo matching using convolutional neural networks. Their commits include modifications to the main training and prediction scripts (`main.lua`, `predict_kitti.lua`), data loading, and preprocessing, as well as implementing supporting CUDA kernels (`adcensus.cu`) and scripts for data handling (`download_middlebury.sh`). They also addressed bugs and made improvements to the overall system, demonstrating a focus on the implementation and performance of the stereo matching models.
convolutional-neural-networks
PyTorch implementation of Barlow Twins.
Role in this project:
ML Engineer Contributions:10 commits, 8 pushes, 34 comments in 9 months
Contributions summary:Jure primarily contributed to the core training and evaluation components of the Barlow Twins implementation. They made changes to the main training loop, including adjusting learning rates and integrating mixed-precision training with `torch.cuda.amp`. Further modifications involved fixing issues related to PyTorch Hub integration and distributed data parallel wrapping in the evaluation script, demonstrating an understanding of model deployment and optimization strategies. Moreover, the user updated the model's pre-trained weights and parameters from available external sources.
pytorch