search

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

How to inference on custom dataset? #5

Closed HalvesChen closed 7 months ago

HalvesChen commented 7 months ago

Hello, thanks for your amazing job. I want to use Semantic Segmentation model finetuned on LoveDA, but it seems that there is no script for inference on custom dataset provided here. How should I do it?

DotWang commented 7 months ago

@HalvesChen mmseg can be used for inference, and we have provided related training and inference commands in README. Before this, you should prepare your dataset as the requirement of mmseg.

SalzingerJ commented 7 months ago

@HalvesChen If that can be useful, I have performed inference based on the author's recommendations in this issue, and left a message at the end to summarise the learnings : https://github.com/ViTAE-Transformer/MTP/issues/2

HalvesChen commented 7 months ago

@HalvesChen If that can be useful, I have performed inference based on the author's recommendations in this issue, and left a message at the end to summarise the learnings : #2

Thanks， your reply is helpful!