facebookresearch / detectron2

Detectron2 is a platform for object detection, segmentation and other visual recognition tasks.
https://detectron2.readthedocs.io/en/latest/
Apache License 2.0
29.3k stars 7.32k forks source link

DensePose的apply_net.py运行dump的选项时候,如何多gpu运行呢? #5279

Open gitPZH opened 1 month ago

gitPZH commented 1 month ago

DensePose的apply_net.py如何加入多gpu运行配置

Programmer-RD-AI commented 1 month ago

Hi, Check the following links it should help you get an understanding and hopefully resolve your issue..

  1. https://discuss.ray.io/t/parallel-detectron2-pytorch-inference-with-gpu/6267
  2. https://detectron2.readthedocs.io/en/latest/tutorials/getting_started.html#training-evaluation-in-command-line
  3. https://github.com/facebookresearch/detectron2/blob/main/projects/DensePose/apply_net.py

Let me know if there is more issues.. Thanks