plantcv icon indicating copy to clipboard operation
plantcv copied to clipboard

Kmeans function changes image size

Open k034b363 opened this issue 8 months ago • 0 comments

Describe the bug Somewhere in the kmeans clustering pipeline, the image size is changing. We think this is a feature of the patch-based approach maybe change dimensions to be divisible by the specified patch size. The proposed fix is to add padding to the right and bottom of the image with some background average value to make sure the patch size is divisible by each dimension, and then slice back to the original dimensions when finished.

To Reproduce

  1. Starting with a set of images, train a model using plantcv.learn.train_kmeans making sure to specify a patch size that doesn't divide evenly into the image pixel dimensions
  2. Using a focal image, classify the pixels in the image using plantcv.predict_kmeans
  3. Check the shape of both the original focal image and the output from the prediction step. If the patch size doesn't divide evenly, the image will be smaller after prediction.

Expected behavior We expect the kmeans function not to change the image size. This is making some downstream checks using the overlay of cluster masks on the original image impossible.

Local environment (please complete the following information):

  • OS: macOS
  • Environment: conda plantcv dev environment
  • PlantCV Version: dev

Additional context Add any other context about the problem here.

k034b363 avatar Jun 17 '24 16:06 k034b363