purbayankar / Hyperspectral-Vision-Transformer

A PyTorch implementation of CNN+Vision Transformer for hyperspectral image classification
64 stars 12 forks source link

How to predict and visualize the classification map? #1

Closed AlanMorningLight closed 3 years ago

AlanMorningLight commented 3 years ago

Thank you very much for sharing. How to predict and visualize the classification map?

purbayankar commented 3 years ago

Thank you very much for sharing. How to predict and visualize the classification map?

Use the "generate_png()" function for generating the classification maps from the model.