pytorch version
An adapted and simplified version of original YOLACT (https://github.com/dbolya/yolact/).
Paper: YOLACT: Real-time Instance Segmentation
Original Yolact authors use pretrained weights of backbone 'resnet-101',which can be download from 'resnet101_reducedfc.pth',
The file 'resnet101_reducedfc.pth'
should be put under './weights/'
directory.
to train a new model with default save folder './weights/'
python train.py
or indicate one
python train.py --save_folder weights/
to train with a weight file
python train.py --resume weights/yolact_base_0_30000.pth