Mrcnn

[LucienZhang]

zip file of basic unit for showing the demo:

lmsvr01:/storage/lucien/coco.zip

essential files used for training on coco dataset:

lmsvr01:/home/zhang/blueoil/lmnet/lmnet/networks/instance_segmentation/annotations/*.json

files in path blueoil/lmnet/lmnet/networks/instance_segmentation/

balloon.py: file to train mask rcnn on ballon dataset.
camera.py: file to run the demo with webcam
coco.py: file to train mask rcnn on coco dataset.
config.py: essential file used for training mask rcnn
keras_linear.py: useless, can be deleted
model.py: useless, can be deleted.
model_quantized.py: quantized model of mask rcnn
mrcnn.py: useless, can be deleted.
parallel_model.py: used if you train the model on multiple GPUs
requirements.txt: the requirements used for this model
resnet18_quantized.py the quantized backbone resnet 18 for the model. run it to train resnet 18 on imagenet or import it as the backbone.
smaller_resnet18_quantized.py: the is a smaller version of resnet 18, not used yet.
utils.py: used in mask r cnn
visualize.py: used for showing the demo.

commands

to train the backbone:
PYTHONPATH=/home/zhang/blueoil/lmnet/ python resnet18_quantized.py
to train mask r cnn on balloon dataset:
PYTHONPATH=/home/zhang/blueoil/lmnet/ python balloon.py train --dataset=/storage2/dataset/balloon --weights=resnet18
to train mask r cnn on coco dataset:
PYTHONPATH=/home/zhang/blueoil/lmnet/ python coco.py train --dataset=/storage/dataset/MSCOCO --model=/home/zhang/blueoil/lmnet/lmnet/networks/instance_segmentation/backbone/q15.hdf5

[CLAassistant]

Thank you for your submission! We really appreciate it. Like many open source projects, we ask that you sign our Contributor License Agreement before we can accept your contribution.
_{You have signed the CLA already but the status is still pending? Let us recheck it.}

[CLAassistant]

Thank you for your submission! We really appreciate it. Like many open source projects, we ask that you sign our Contributor License Agreement before we can accept your contribution.
_{You have signed the CLA already but the status is still pending? Let us recheck it.}