search

matej-ulicny / harmonic-networks

BSD 3-Clause "New" or "Revised" License
57 stars 7 forks source link

Model training for instance segmentation #2

Open animesh-007 opened 3 years ago

animesh-007 commented 3 years ago

Can you guide me, how to use harmonic network for instance segmentation using this code?

matej-ulicny commented 3 years ago

The instance segmentation is built upon mmdetection framework v0.6.0. To run the code the mmdet package has to be built. The best is to create a virtual enironment with pytorch (at the time of development it was v1.1.0). The package is installed by these instructions:

pip install mmcv==0.2.8
cd mmdetection
./compile.sh
python setup.py develop

The config files for training are adapted for COCO dataset which has to be placed in data/ folder as described in mmdetection 

./tools/dist_train.sh ${CONFIG_FILE} ${NUM_GPUS} [--validate]