Skip to content
Snippets Groups Projects
Commit 8c86f74c authored by Kamran Melikov's avatar Kamran Melikov Committed by Kai Chen
Browse files

Non color images (#1976)

* First Draft

On branch non-color-images
Changes to be committed:
modified:   mmdet/datasets/pipelines/loading.py

* Add option to load non color images

Add 'color_type' parameter to LoadImageFromFile class
Change __repr__ method accordingly
Since non-color images maybe two dimensional expand image
dimensions if necessary in DefaultFormatBundle and
ImageToTensor classes

Changes to be committed:
    modified:   mmdet/datasets/pipelines/formating.py
    modified:   mmdet/datasets/pipelines/loading.py

* Fix RandomCrop to work with grayscale images

Changes to be committed:
    modified:   mmdet/datasets/pipelines/transforms.py

* Modify retrieving w, h of padded image in anchor heads

This addreses problems with single channel images for which the
shape is  tuple with 2 values

Changes to be committed:
    modified:   mmdet/models/anchor_heads/anchor_head.py
    modified:   mmdet/models/anchor_heads/guided_anchor_head.py
    modified:   mmdet/models/anchor_heads/reppoints_head.py
parent e60d34af
No related branches found
No related tags found
No related merge requests found
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment