Training Pipeline - Githubissues

havanagrawal / clomask

Capstone Project for Clobotics: Using Mask R-CNN for Rigid/Non-Rigid Retail Consumable Product Detection

MIT License

7 stars 4 forks source link

Closed pshivraj closed 5 years ago

pshivraj commented 5 years ago

This PR is an end to end pipeline for model Training. It includes:

Data pre-processing using pre_process.py to generate .h5 file for masks.

Custom Config for clomask Dataset. (config.py)

Model Training using coco and imagenet weights. (train.py)

This caters to following issues: #18 #17 #26