Jahanzeb Rathore

in the prediction.py file, the probs dimension in my implementation is (5,256,256) which incites the error of channels>4 not allowed. I get it that its the .toPILImage function as it...

No validation on Custom dataset

1

I am using the following config ###### mmdet_base = "../../thirdparty/mmdetection/configs/_base_" _base_ = [ f"{mmdet_base}/models/faster_rcnn_r50_fpn.py", f"{mmdet_base}/datasets/coco_detection.py", f"{mmdet_base}/schedules/schedule_1x.py", f"{mmdet_base}/default_runtime.py", ] data_root = 'data/mitosis_coco/' classes = ('mitosis',) pretrained = 'https://github.com/SwinTransformer/storage/releases/download/v1.0.0/swin_tiny_patch4_window7_224.pth' # noqa model...