The cnn.py implemented a simple CNN with pytorch. The network consists of two convolutional-ReLU-pooling layer and a fully-connected-softmax layer. The network structure is as following: ...
Official Pytorch Code base for "MobileUtr: Revisiting the relationship between light-weight CNN and Transformer for efficient medical image segmentation" ...
Add a description, image, and links to the machine-learning-computer-vision-pytorch-cnn topic page so that developers can more easily learn about it.
A PyTorch implementation of the CVPR 2018 paper Domain Adaptive Faster R-CNN for Object Detection in the Wild. The original code used by the authors can be found here. This implementation is built on ...
A deep learning project implementing a ResNet-based Convolutional Neural Network for classifying food images from the Food-101 dataset. This project demonstrates state-of-the-art computer vision ...
A real-time facial emotion recognition web app using a custom-trained CNN on the FER-2013 dataset. Built with Flask, OpenCV, and PyTorch.