ying09 / TextFuseNet

A PyTorch implementation of "TextFuseNet: Scene Text Detection with Richer Fused Features".
MIT License
476 stars 123 forks source link

How i can build detectron2 with cuda 10.0 or cpu #4

Closed tanphuc16797 closed 4 years ago

tanphuc16797 commented 4 years ago

I have some problems when i tried to run python demo/icdar2013_detection.py on pytorch 1.4 - cuda 10.0

Real-YeJ commented 4 years ago

@tanphuc16797
For the first problem, our TextFuseNet is based on the early detectron2, which needs cuda ==10.1 and pytorch >= 1.3, so I suggest that you install it according to our installation. if you want to use pytorch 1.4 and cuda 10.0, please make sure they are suitable. The reason of KeyError is that you didn't build TextFuseNet by running 'python setup.py build develop'.

For the second problem, our implementation needs gpu. Maybe you can find how to building detectron2 with cpu in https://github.com/facebookresearch/detectron2