This is a stripped down version of The FAIR Mask R-CNN Implementation. There were some problems with using the jupyter notebooks and Keras multiprocessing on Windows-10 64bit machines.
-
Clone this repo into a fresh python 3.6 virtual environment.
git clone https://github.com/FletcherFT/mrcnn-playground.git
-
Install the requirements.txt file.
pip install -r requirements.txt
Note If you receive an error from pip when trying to install the
shapely
module, install it manually either from here, or using the .whl included in this repo:pip install .\extras\Shapely-1.6.4.post2-cp36-cp36m-win_amd64.whl
Then try the first command in this step again. Hopefully it should work on your system :)
-
(opt.) Install the optional-requirements.txt file, which contains the tensorflow-gpu package.
pip install -r optional-requirements.txt
-
Test the training script:
python train_shapes.py