issues
search
zjhuang22
/
maskscoring_rcnn
Codes for paper "Mask Scoring R-CNN".
MIT License
1.9k
stars
378
forks
source link
Add two augmentation functions: rotate and light change and more loss weights
#80
Open
qizhen816
opened
4 years ago
qizhen816
commented
4 years ago
Bounding box and polygon/mask rotate
Using OpenCV getRotationMatrix2D to get rotaton matrix from the angle given by config files.
Bounding box points are calculated as well as polygon instance points.
before:
after(-20 degree):
Image vertical/horizontal linear light change
Given a linear light interval limit L, then the image light will be linearly adjusted within [-L, L].
The function is separated to work vertically or horizontally.
before:
after(L=100, vertical):
after(L=100, horizontal):
More Loss weight: MASK_LOSS_WEIGHT BOX_LOSS_WEIGHT CLASSIFICATION_LOSS_WEIGHT and MASKIOU_LOSS_WEIGHT
All 4 kinds of loss weights can be altered in \maskrcnn_benchmark\config\defaults.py