Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

How to modify the conv1 of the pre training weights of the Mask RCnn to fit my 5-channel dataset #11687

Open
MoonDiana17 opened this issue May 7, 2024 · 0 comments
Assignees
Labels
reimplementation Issues in model reimplementation

Comments

@MoonDiana17
Copy link

Describe the issue

In order to increase the number of channels to 5, I have made some modifications to the code. Until this step, I knew that the conv1 of the pre training weight did not match the number of channels in the input image, but I didn't know how to correct it. Even if I set init_cfg=None, I will still report an error.

Reproduction

  1. What command or script did you run?
python tools/train.py work_dirs/mask-rcnn_r101-caffe_fpn_ms-poly-3x_coco/mask-rcnn_r101-caffe_fpn_ms-poly-3x_coco.py

Snipaste_2024-05-07_17-31-56

@MoonDiana17 MoonDiana17 added the reimplementation Issues in model reimplementation label May 7, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
reimplementation Issues in model reimplementation
Projects
None yet
Development

No branches or pull requests

2 participants