voxel51 / fiftyone

Refine high-quality datasets and visual AI models
https://fiftyone.ai
Apache License 2.0
8.82k stars 557 forks source link

[FR] Draw bounding box/segmentation mask heatmap of a dataset #1087

Open suvojit-0x55aa opened 3 years ago

suvojit-0x55aa commented 3 years ago

Proposal Summary

Option to draw bounding box/segmentation mask heatmap of a dataset, and see how the the objects are distributed spatially.

Motivation

It's easier to identify the regions in which objects are

Has to be done manually.

What areas of FiftyOne does this feature affect?

Details

(Use this section to include any additional information about the feature. If you have a proposal for how to implement this feature, please include it here.)

Willingness to contribute

The FiftyOne Community encourages new feature contributions. Would you or another member of your organization be willing to contribute an implementation of this feature?

brimoor commented 3 years ago

Hi @suvojit-0x55aa 👋

This sounds like a useful feature to have. My proposal would be that this starts as a method in a new module within the fiftyone.utils subpackage like this:

def visualize_object_locations(sample_collection, label_field, ...):
    ...

which would return the heatmap data and/or plot it directly. If features like this become heavily used, we could support generating + viewing such heatmaps directly in the App.

Would you be able to take a shot at contributing a PR that gets this feature off the ground? 😄

daniel-merrick commented 1 year ago

Was this ever created?