search

liruilong940607 / Pose2Seg

Code for the paper "Pose2Seg: Detection Free Human Instance Segmentation" @ CVPR2019.
http://www.liruilong.cn/projects/pose2seg/index.html
MIT License
532 stars 136 forks source link

Update build_model.py #49

Closed atulchoudhary29 closed 11 months ago

atulchoudhary29 commented 11 months ago

By default, it takes Cuda. But if cuda is not supported on GPU, it switches to CPU. So, I added the line for this purpose.