tanxuehan/segment

segmentation

Quick start

Prerequisites

• Sklearn
• OpenCV for Python
• Pytorch ( recommended version : 1.7.1 )

Installation

cd ./segment
make install

Train

python3 train.py

Test

python3 -m segment.eval model_path eval_res_path

Introduction

Metrics

Backbone	Dice coefficient	Model
Vgg16_BN	27.82%	./weights/mode_split_best_coeff_0.278169_epoch_24.pth

Visualized results

Details

link