ViTAE-Transformer / MTP

The official repo for [JSTARS'24] "MTP: Advancing Remote Sensing Foundation Model via Multi-Task Pretraining"
MIT License
175 stars 11 forks source link

关于旋转目标检测 #23

Open lzJune opened 3 months ago

lzJune commented 3 months ago

您好作者,如果我想用您给的预训练模型训练我自己的数据集的话,有两个问题想要请教您 1.因为我的数据集是细粒度类别的目标检测,所以想用您FAIR1M训练的config进行训练,是不是用

pretrained ='/work/share/achk2o1zg1/diwang22/work_dir/multitask_pretrain/pretrain/avg/with_background/vit_b_rvsa_224_mae_samrs_mtp_three/last_vit_b_rvsa_ss_is_rd_pretrn_model_encoder.pth' 把这个预训练模型下载下来然后训练就可以了? 2.看您论文里没提多尺度训练的事,需要进行多尺度的切割吗比如[0.5,1.0,1.5],还是就单尺度切割就可以了呢?

希望您能回复,谢谢!

DotWang commented 3 months ago

@lzJune

  1. 是的
  2. FAIR1M刷榜默认用多尺度裁剪