* add solo style dicecost
* deiceloss
* add dicecost that knet used by set naive_dice to False to use it
* fix test_assigner config
* add some description of the naive_dice of DiceCost and remove MaskCost which is same to CrossEntropyLossCost
* add some description of the naive_dice of DiceCost and remove MaskCost which is same to CrossEntropyLossCost
* add test for dicecost that naive_dice is False
* add test for dicecost that naive_dice is False
* remove the test of no gt masks for the naive_dice option of DiceCost is False
* remove the test of no gt masks for the naive_dice option of DiceCost is False
* remove the test of no gt masks for the naive_dice option of DiceCost is False
* add test of DiceCost that naive_dice is False
* add a discribition of why we use the shape of mask_pred
* add a discribition of why we use the shape of mask_pred
Co-authored-by: mengwg <976537406@qq.com>
Co-authored-by: fms <2635540050@qq.com>
* Convert valid_pred_scores to float32 in sim_ota_assigner
Workaround to resolve `RuntimeError: "sqrt" "_vml_cpu" not implemented for 'Half'`
* Add unit test for SimOTAAssigner
* Cast output of binary_cross_entropy back to float16 if needed
Also fix yapf lint issue.
* Lint for yapf
* More lint fixes on test_assigner.py
* Cast cls_scores back to float16 directly
Co-Authored-By: Wenwei Zhang <ZwwWayne@users.noreply.github.com>
Co-authored-by: Wenwei Zhang <ZwwWayne@users.noreply.github.com>
update doc
update doc format
deepcopy pixel_decoder cfg
move mask_pseudo_sampler cfg to config file
move part of postprocess from head to detector
fix bug in postprocessing
move class setting from head to config file
remove if else
move mask2bbox to mask/util
update docstring
update docstring in result2json
fix bug
update class_weight
add maskformer_fusion_head
add maskformer fusion head
update
add cfg for filter_low_score
update maskformer
update class_weight
update config
update unit test
rename param
update comments in config
rename variable, rm arg, update unit tests
update mask2bbox
add unit test for mask2bbox
replace unsqueeze(1) and squeeze(1)
add unit test for maskformer_fusion_head
update docstrings
update docstring
delete \
remove modification to ce loss
update docstring
update docstring
update docstring of ce loss
update unit test
update docstring
update docstring
update docstring
rename
rename
add msdeformattn pixel decoder
maskformer refactor
add strides in config
remove redundant code
remove redundant code
update unit test
update config
update
* first commit
* add README
* move model description from config to readme
add description for binary_input
add description for dice loss
add a independent panoptic gt processing function
add a independent panoptic gt processing function
remove compatibility of pretrain in maskformer
* update comments in maskformer_head
* update docs format
* [Feature] Support TOOD.
* update
* use assign result
* use assign result
* clean assigner
* add config
* add tood head unit test and fix device bug
* test assigner and fix empty gt error
* test hook
* add anchor-based cfg and readme
* update readme
* resolve comments
* resolve comment
* add metafile
* fix model index
* copyright
* resolve comments
* resolve comments
* Add YOLOF inference
* Update
* Update Uniform match
* Update param
* Update cfg
* Fix bug about assigner
* Add reduce_mean
* update yolof
* update docstr
* add iter base config
* update code and add docstr
* fix comment and update docstr
* update transforms and add docstr
* Simplify the code.
* Enhancement random_shift and add unit test
* add rest of unit test and process empty gt.
* add README.md
* update comment
* Fix unittest
* update model link
* supports for DETR transformer
* unit test with small data to avoid out of memory in CI
* use batch size 1 for unit test to avoid out of memory
* move transformer into utils folder and use more small data for unit test
* reformat docstring
* add more detailed docstring
* reforamt
* reformat and add build_transformer (#3866)
* Supports for DETR position embedding (#3850)
* supports for DETR position embedding
* reformat docstring
* reformat
* move transformer_head folder into utils
* remove unnecessary comments
* reformat
* reformat
* reformat docstring
* reformat
* add optional for default parameters
* add comments for unit test
* avoid inplace in MultiheadAttention and add optional add_residual for FFN
* Supports for DETR inference (#3941)
* fix forward in FFN
* supports detr inference mode with batchsize 1
* add docstring for DETR in detr.py
* reformat and add doctest for TransformerHead
* reformat, and move the mask construction from simple_test in detr.py to the forward process of transformer_head
* rename detr_r50_1x_coco as detr_r50_8x4_1x_coco in config
* reformat
* mask input_img_shape in img_meta as a general modification of base_detector forward_test
* reformat
* Supports for DETR hungarian matcher. (#3929)
* Add doc of modify loss (#3777)
* add doc of modify loss
* minor fix
* change name
* improve the function of simple_test_bboxes (#3853)
* update
* fix lint error
* add bbox_pred_split in sabl_head
* fix lint error
* fix yapf
* fix build error
* fix yapf
* Clean background_labels in the dense heads (#3221)
* Clean background_labels in the dense heads
* clean format
* Reformat
* Fix CI bug
* fix fsaf and the comments
* fix newly added heads
* fix yolact
* complete comments
* fix rpn transforming bug in two stage networks (#3754)
* fix rpn transforming bug in two_stage
* [Refactor] refactor get_subset_by_classes in dataloader for training with empty-GT images (#3695)
* Add regression test and test data
* Add fix for empty gt images
* Trigger CI build
* refactor get_subset_by_classes
* reformat and fix docstring
* [refactor]: move get_subset_by_classes to _filter_imgs
* make img_ids consistent
* resolve comments
* simplify logics
* add warning
* add warning
* add warning
Co-authored-by: mmeendez8 <miguelmndez@gmail.com>
* Fix nonzero in NMS for PyTorch 1.6.0 (#3867)
* Support TTA of ATSS, FCOS, YOLOv3 (#3844)
* Support TTA of ATSS, FCOS, YOLOv3
* Add comment
* [Docs] Fix typo in docs/tutorials/new_dataset.md (#3876)
* [Docs] Remove duplicate content in docs/config.md (#3875)
* [Enhance]: Convert mask to bool before using it as img's index for robustness and speedup (#3870)
* Convert mask to bool before using it as img's index
* lint
* Rewrite code in more concise way
* Fix typo in bbox_flip (#3886)
* fix the API change bug of PAA (#3883)
* fix cv2 import error of ligGL.so.1 (#3891)
* [enhance]: Improve documentation of modules and dataset customization (#3821)
* update tutorials for changing training settings
* update
* Update details of hook
* resolve comments
* reformat
* Update module documentations
* add notes for classes specification
* support to use pytorch 1.6 in docker (#3905)
* support to use pytorch 1.6 in docker
* update mmcv requirement
* update docs
* Add missing notes in data customization (#3906)
* [Fix]: fix mask rcnn training stuck problem when there is no positive rois (#3713)
* Fix mask rcnn stuck problem when there is no positive rois
* support non pos inference in cascade methods, link CU-49tawu
* print mmcv version in CI
* use mmcv repo to check wrappers
* change cpu build
* upgrade mmcv requirements and change ci back
* use pre-built whl in CI
* Bump to v2.5.0 (#3879)
* bump to v2.5.0
* update PR
* resolve comments
* clean unnecessary modification
* update bc breaking
* rephrase bc-breaking
* rephrase
* Added `generate_inputs_and_wrap_model` function for pytorch2onnx (#3857)
* added `generate_inputs_and_wrap_model` function for pytorch2onnx
* undo debugging changes
* added model.cpu().eval()
* refactored onnx
* fix verify
* move `generate_inputs_and_wrap_model` into mmdet.core.export
* use `generate_inputs_and_wrap_model` in tools/pytorch2onnx.py
* typo (#3917)
* Edit mmdet.core.export docstring (#3912)
* edit mmdet.core.export docstring
* update doctest
* fix small bugs
* supports for HungarianMatchAssigner, add bbox_cxcywh_to_xyxy and bbox_xyxy_to_cxcywh for bbox transform
* format box-wise related giou calculating as a function and implement it in iou2d_calculater.py
* supports for BboxGIoU2D and re-implements giou_loss using bbox_gious
* remove unnecessary
* reformat
* reformat docstring
* reformat
* rename
* supports for giou calculating in BboxOverlaps2D, and re-implements giou_loss using bbox_overlaps
* fix sabl validating bug in cascade_rcnn (#3913)
* reformat
* move giou related unit test from test_losses.py to test_iou2d_calculator.py
* reformat
* Avoid division by zero in PAA head when num_pos=0
* [Fix]: Avoid division by zero in PAA head when num_pos=0 (#3938)
* explicitly add mode in giou_loss
* Add supports for giou calculation in BboxOverlaps2D, and re-implements giou_loss using bbox_overlaps (#3936)
* supports for BboxGIoU2D and re-implements giou_loss using bbox_gious
* remove unnecessary
* reformat
* reformat docstring
* reformat
* rename
* supports for giou calculating in BboxOverlaps2D, and re-implements giou_loss using bbox_overlaps
* reformat
* move giou related unit test from test_losses.py to test_iou2d_calculator.py
* reformat
* Avoid division by zero in PAA head when num_pos=0
* explicitly add mode in giou_loss
* Add supports for giou calculation in BboxOverlaps2D, and add iou_calculator in hungarian_assigner
* rename hungarian_match_assigner as hungarian_assigner
* fix init
* reformat docstring
* Avoid division by zero in PAA head when num_pos=0
* fix cpu (#3948)
* add mode for iou_calculator and make giou cost as a default case
* make mode as a param in iou_calculator
* reformat docsting
* make iou_mode outside of iou_calculator
Co-authored-by: Cao Yuhang <yhcao6@gmail.com>
Co-authored-by: yuzhj <31239926+yuzhj@users.noreply.github.com>
Co-authored-by: Wenwei Zhang <40779233+ZwwWayne@users.noreply.github.com>
Co-authored-by: mmeendez8 <miguelmndez@gmail.com>
Co-authored-by: Yosuke Shinya <42844407+shinya7y@users.noreply.github.com>
Co-authored-by: LabMemNo003 <LabMemNo003@gmail.com>
Co-authored-by: Ye Liu <yeliudev@outlook.com>
Co-authored-by: aboettcher <aboettcher@users.noreply.github.com>
Co-authored-by: Ryan Li <xlide@connect.ust.hk>
Co-authored-by: Jiangmiao Pang <pangjiangmiao@gmail.com>
* Re-implements RandomCrop to support different crop_type (#4093)
* re-implements RandomCrop to support different crop_type
* add assertion for relative related crop_type
* reformat, and add image shape assertion for unit test
* reformat
* Supports for DETR training mode in process. (#3963)
* add input_img_shape in img_meta, and supports training mode toy
* toy commit
* toy commit
* re-organize and support for detr training mode
* add unit test for detr training, testing, and transformer_head
* add more unit test for transformer
* reformat
* remove unnecessary
* use small transformer model for unit test
* reformat docstring
* reformat
* reformat
* Supports DETR e150 config (#4197)
* rename e300 config as e150 config
* reformat
* add comments on override option in Resize
* add comments on override option in Resize
* position embeddingto positional encoding
* fix unit tests
* fix registry name bug
* rename file
* remove unnecessary
Co-authored-by: ZwwWayne <wayne.zw@outlook.com>
Co-authored-by: Cao Yuhang <yhcao6@gmail.com>
Co-authored-by: yuzhj <31239926+yuzhj@users.noreply.github.com>
Co-authored-by: Wenwei Zhang <40779233+ZwwWayne@users.noreply.github.com>
Co-authored-by: mmeendez8 <miguelmndez@gmail.com>
Co-authored-by: Yosuke Shinya <42844407+shinya7y@users.noreply.github.com>
Co-authored-by: LabMemNo003 <LabMemNo003@gmail.com>
Co-authored-by: Ye Liu <yeliudev@outlook.com>
Co-authored-by: aboettcher <aboettcher@users.noreply.github.com>
Co-authored-by: Ryan Li <xlide@connect.ust.hk>
Co-authored-by: Jiangmiao Pang <pangjiangmiao@gmail.com>
* YOLACT (not the final commit, so please don't review)
* Implementation notes for YOLACT
* Modify yolact_r50.py
* Update YOLACT configs
* Add YOLACT README
* Adjust according to the first round of review
* Handle the corner case when gt_mask is empty
* Add unit tests for YOLACT heads
* Add unit tests for YOLACT forward
* Adjust according to Kai's comments, except for gt_mask filtering
* Change use_OHEM to use_ohem in unit tests
* Update config in YOLACT unit tests
* merge bbox overlap and batch bbox overlap
* fix unit test
* refactor some codes and fix some bugs
* fix isort
* fix ci
* update model link
* fix cfg path
* support batch test
* Make get_seg_masks faster
* fix format
* update readme
Co-authored-by: Cao Yuhang <yhcao6@gmail.com>