InternImage icon indicating copy to clipboard operation
InternImage copied to clipboard

TRAIN ON CUSTOM DATASET | SEGMENTATION

Open EthanAbitbol3 opened this issue 1 year ago • 4 comments

Hello and thank you for sharing your work, i was wondering if you woul dprovide some information about training on custom dataset ?

Thank you !

EthanAbitbol3 avatar Jun 21 '23 14:06 EthanAbitbol3

You can look at mmsegmentation 0.27 repo demo. You'll see how to set up a custom dataset there

fschvart avatar Jun 22 '23 17:06 fschvart

I appreciate your help! Prior to attempting the method you suggested, I also made an effort to retrain using the ADE20k dataset to evaluate its performance. Regrettably, it did not yield the desired results.

Here is the command I executed: !set CUDA_VISIBLE_DEVICES=0 !python train.py configs/ade20k/upernet_internimage_t_512_160k_ade20k.py

However, I encountered the following error: image

I proceeded to make the following modifications to the upernet_internimage_t_512_160k_ade20k.py file: image

Additionally, I made the following changes to the ade20k.py file: image

I encountered the same problem again with the cfg.dump function, which caused the issue to persist. image

NB : I am on Windows.

EthanAbitbol3 avatar Jun 23 '23 10:06 EthanAbitbol3

Shalom @EthanAbitbol3 ,

Even though you're on Windows, use relative reference and '/' instead of '' or '\'

fschvart avatar Jun 23 '23 16:06 fschvart

Shalom! Thank you for your response. I have made the changes as you suggested. image image

Unfortunately, I encountered the same error once again.

EthanAbitbol3 avatar Jun 26 '23 07:06 EthanAbitbol3