THU-MIG / RepViT

RepViT: Revisiting Mobile CNN From ViT Perspective [CVPR 2024] and RepViT-SAM: Towards Real-Time Segmenting Anything
https://arxiv.org/abs/2307.09283
Apache License 2.0
681 stars 55 forks source link

repvit-sam train self built datasets and an error #25

Open rememberBr opened 7 months ago

rememberBr commented 7 months ago

I have two questions:

  1. Will repvit-sam provide a pipeline for training self built datasets? Make it only segment the categories we need, not anything else
  2. I used tim=0.9.12 and run: Python app\app.py, on the 117 line of timm\models_factory.py will throw an error: TypeError: revit() got an unexpected keyword argument 'pretrained' Cfg ' timm\models_factory.py Line 117: Model=create Fn( Pretrained=pretrained, Pretrained_Cfg=pretrained_Cfg, Pretrained_Cfg_Overlay=pretrained_Cfg_Overlay, **Kwargs, ) if remove Pretrained_Cfg_Overlay and Pretrained_Cfg, I can successfully run it
jameslahm commented 7 months ago

Thanks for your interest.

Will repvit-sam provide a pipeline for training self built datasets? Make it only segment the categories we need, not anything else

We will release the code for training self built datasets.

I used tim=0.9.12 and run: Python app\app.py, on the 117 line of timm\modelsfactory.py will throw an error: TypeError: revit() got an unexpected keyword argument 'pretrained' Cfg '

timm==0.5.4 should be fine.

rememberBr commented 7 months ago

Looking forward to your work and hoping to see further updates as soon as possible