# PyTorchCV

**Repository Path**: gufan1988/PyTorchCV

## Basic Information

- **Project Name**: PyTorchCV
- **Description**: Source code for most of deep learning based cv problems by PyTorch.
- **Primary Language**: Unknown
- **License**: Apache-2.0
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 1
- **Created**: 2020-06-29
- **Last Updated**: 2025-08-23

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

### 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](https://github.com/youansheng/pytorch-cv/tree/master/methods/pose)
    - 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](https://github.com/youansheng/pytorch-cv/tree/master/methods/det)
    - SSD: Single Shot MultiBox Detector
    
- [Semantic Segmentation](https://github.com/youansheng/pytorch-cv/tree/master/methods/seg)
    - Efficient ConvNet for Real-time Semantic Segmentation

- [Image Classification](https://github.com/youansheng/pytorch-cv/tree/master/methods/cls)

#### Examples
- Train the openpose model
```bash
python main.py --hypes hypes/pose/coco/op_coco_pose.json \
               --base_lr 0.001 \
               --phase train
```

- Finetune the openpose model
```bash
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):
```bash
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):
```bash
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:
```bash
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.