ShuLiu1993 / PANet

PANet for Instance Segmentation and Object Detection
MIT License
1.34k stars 281 forks source link

Inference on CPU #39

Open ashnair1 opened 5 years ago

ashnair1 commented 5 years ago

Is it possible to run inference on CPU? In the forward function of the roi_Xconv1fc_gn_head_panet in fast_rcnn_heads.py, it relies on the gpu version of the roi align. How can this issue be solved?