Can the pre training weights be cancelled？

JCruan519 / VM-UNet

(ARXIV24) This is the official code repository for "VM-UNet: Vision Mamba UNet for Medical Image Segmentation".

Apache License 2.0

457 stars 17 forks source link

Can the pre training weights be cancelled？ #68

Open JacCubk opened 1 month ago

JacCubk commented 1 month ago

Hello, thank you for providing such a great job.

I am trying to use this work for processing hyperspectral images, but I would like to cancel the use of pre trained weights as they are applicable to the 3-channel RGB images. May I ask if it's convenient for you to let me know how I should modify the code?

Thank you very much!

lwtgithublwt commented 1 month ago

您好，感谢您提供如此出色的工作。

我正在尝试使用这项工作来处理高光谱图像，但我想取消使用预训练权重，因为它们适用于 3 通道 RGB 图像。请问您让我知道我应该如何修改代码是否方便？

谢谢！给个none不就行了