ztwe/pytorch-cv

Repo for Image Classification, Object Detection, Segmentation & Pose Estimation.

Source code for CV Problems(Pytorch)

@misc{CV2018,
  author =       {Donny You (youansheng@gmail.com)},
  howpublished = {\url{https://github.com/youansheng/pytorch-cv}},
  year =         {2018}
}

This repository provides source code for some cv problems. We do our best to keep this repository up to date. If you do find a problem about this repository, please raise this as an issue. We will fix it immediately.

Details:

• Pose Estimation

  • Convolutional Pose Machines
  • Realtime Multi-Person 2D Pose Estimation using Part Affinity Fields
  • Associative Embedding: End-to-End Learning for Joint Detection and Grouping

• Object Detection

  • SSD: Single Shot MultiBox Detector

• Semantic Segmentation

  • Efficient ConvNet for Real-time Semantic Segmentation

• Image Classification

Examples

• Train the openpose model

python main.py --hypes hypes/pose/coco/op_coco_pose.json \
               --base_lr 0.001 \
               --phase train

• Finetune the openpose model

python main.py --hypes hypes/pose/coco/op_coco_pose.json \
               --base_lr 0.001 \
               --phase train \
               --resume checkpoints/pose/coco/coco_open_pose_65000.pth

• Test the openpose model(test_img):

python main.py --hypes hypes/pose/coco/op_coco_pose.json \
               --phase test \
               --resume checkpoints/pose/coco/coco_open_pose_65000.pth \
               --test_img val/samples/ski.jpg

• Test the openpose model(test_dir):

python main.py --hypes hypes/pose/coco/op_coco_pose.json \
               --phase test \
               --resume checkpoints/pose/coco/coco_open_pose_65000.pth \
               --test_dir val/samples

• Create the submission:

python main.py --hypes hypes/pose/coco/op_coco_pose.json \
               --phase submission \
               --resume checkpoints/pose/coco/coco_open_pose_65000.pth \
               --test_dir coco_test_dir

• Attention: Other command line parameters are showed in main file. You can refer & use them.