Skip to content

Classifying With a Score - Conceptual Question #2097

Answered by AdeelH
natedailey7 asked this question in Q&A
Discussion options

You must be logged in to vote

RV does have some support for regression but it is limited to chip-level regression i.e. predicting a single number for a full chip, rather for each pixel in the chip.

In your case, you could subclass the semantic segmentation learner and dataset classes into RasterRegression* classes with some modifications. You would need to modify the loss function from cross-entropy to MSE or MAE, and modify the train_step and valid_step methods (see RegressionLearner), and also postprocess_model_output to not apply softmax. The GeoDataset class will have to be modified too.

So it will take some work, but if you do manage to get something working, I'd appreciate a PR!

Replies: 1 comment 2 replies

Comment options

You must be logged in to vote
2 replies
@natedailey7
Comment options

@AdeelH
Comment options

Answer selected by natedailey7
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Category
Q&A
Labels
None yet
2 participants