Skip to content

My personal repo for templates for image recognition with PyTorch

Notifications You must be signed in to change notification settings

ali-yz/image-recognition

Repository files navigation

Image Recognition

Here I tried to make a set of templates for different tasks in image recognition and computer vision using the TorchVision package.

If your new to Deep Learning

  1. PyTorch-Tutorials

Tasks

  1. Binary Classification (CNN from scratch, Sigmoid Activation, BinaryCrossEntropyLoss)
  2. MultiClass Classification (CNN from scratch, ReLu, CrossEntropy)
  3. MultiClass Classification (CNN from pre-trained models)
  4. Object Detection (find object label, position, mask)

Usage

Read instruction in each folder's README file for each task's acceptable input data format.

About

My personal repo for templates for image recognition with PyTorch

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published