Skip to content

Latest commit

 

History

History
67 lines (55 loc) · 2.31 KB

README.md

File metadata and controls

67 lines (55 loc) · 2.31 KB

Source code for CV Problems(Pytorch)

@misc{CV2018,
  author =       {Donny You ([email protected])},
  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:

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.