Helmholtz-AI-Energy / TBBRDet

Thermal Bridges on Building Rooftops Detection (TBBRDet)
BSD 3-Clause "New" or "Revised" License
16 stars 6 forks source link

Request for Pretrained Weight File for Swin Transformer on Mask R-CNN #5

Open aliwaqas333 opened 3 months ago

aliwaqas333 commented 3 months ago

First of all, I would like to thank you for your outstanding work on integrating Swin Transformer with Mask R-CNN and providing the valuable resources and codebase to the community.

I have encountered some difficulties in achieving satisfactory training accuracy. My results are considerably lower than the ones in your paper. Here are a few key metrics from my training results:

Bounding Box Detection (bbox):

Segmentation (segm):

My training configuration is as follows:

PYTHONFILE=scripts/mmdet/train.py
CONFIG_FILE=configs/mmdet/swin/mask_rcnn_swin-t-p4-w7_fpn_ms-crop-3x_coco.py

python ${PYTHONFILE} ${CONFIG_FILE} --work-dir ${WORKDIR}

The availability of the pretrained weight file would greatly enhance the reproducibility of your results.