sailfish009 / fcos.pytorch Goto Github PK

• ctr. on reg
• giou loss
• ctr. sampling

• normalizing the regression targets

• opencv-python
• pytorch >= 1.2
• torchvision >= 0.4.

Let's say the white boxes are the gt boxes, the points of different colors represent the sampling points of different feature layers while applying ctr-sampling.

Train on 2 1080Ti, 3 imgs for each gpu, init lr=1e-5 cosine decays to 1e-6, but performance is not good on VOC07test. Maybe should remove centerness head while applying central sampling.

Due to computational resource constraints, I was unable to fully train the model on the COCO dataset. I have converted the official pre-training model weights FCOS_R_50_FPN_1x into my own.
The converted weights is avaliable Baidu driver link, password: rpni
The official implementation of preprocessing(pixel is not normalized to 0-1 and input img follows BGR fomat ) is a little different from mine.

some excellent work based on this repo:
FCOS-Pytorch-37.2AP
FCOS_DET_MASK