-
Ouail authored
* add a FORCE_CUDA flag Following discussion [here](https://github.com/facebookresearch/maskrcnn-benchmark/issues/167), this seemed the best solution * Update Dockerfile * Update setup.py * add FORCE_CUDA as an ARG * modified: docker/Dockerfile modified: setup.py * small fix to readme of demo * remove test print * keep ARG_CUDA * remove env value and use the one from ARG * keep same formatting as source * change proposed by @miguelvr * Update INSTALL.md
90c226cf
Name |
Last commit
|
Last update |
---|---|---|
.. | ||
Mask_R-CNN_demo.ipynb | ||
README.md | ||
demo_e2e_mask_rcnn_R_50_FPN_1x.png | ||
demo_e2e_mask_rcnn_X_101_32x8d_FPN_1x.png | ||
predictor.py | ||
webcam.py |