/deep-learning-works

my personal work on deep learning using pytorch

Primary LanguageJupyter Notebook

deep-learning-works

Work in deep learning via Pytorch.

Installation

  • CUDA 9.2
  • cudnn 7.3.1
  • inplace-abn
  • Pytorch 1.4.0, torchvision 0.5.0
    pip install torch==1.4.0+cu92 torchvision==0.5.0+cu92 -f https://download.pytorch.org/whl/torch_stable.html
    
  • Cython
    pip install Cython
    
  • python packages in requirement.txt

Overview

Fork from following task

Person Re-identification

Pedestrian Attribute Recognition

  • Multitask Learning using fully connected layer

Object Detection

Clothing Detection, Keypoints

Result

working on it