search

last-one / Pytorch_Realtime_Multi-Person_Pose_Estimation

Pytorch version of Realtime Multi-Person Pose Estimation project
MIT License
216 stars 70 forks source link

Could you tell me what masklist/train2017.txt ? #6

Open ghost opened 6 years ago

ghost commented 6 years ago

Sorry I run preproccesing/generate_json_mask.py, but masklist/train2017.txt was not generated. What is the file ?

And I have no multi gpu but one gpu, should I remove line 49 "model = torch.nn.DataParallel(model, device_ids=args.gpu).cuda()" in train_pose.py ?

last-one commented 6 years ago

The mask has been generated. You can generate the file by youself.

And you can remove this line.

ghost commented 6 years ago

Thank you for replying ! Is mask_dir masklist/train2014.txt's path ?

tztztztztz commented 6 years ago

I have the same question. I just got a directory which contains a lot of .npy file, rather than a single .txt mask file? @last-one

last-one commented 6 years ago

It needs to be generated by yourself.

tztztztztz commented 6 years ago

@last-one thanks for your reply. fixed it

wangfei-cs commented 6 years ago

@last-one You said 'The mask has been generated. You can generate the file by youself.' sorry, but how do i generate the file by myself?

last-one commented 6 years ago

@wangfei-cs please see README

last-one commented 6 years ago

I have added the code for generating masklist/train2017.txt. Please see the script