Note
Go to the end to download the full example code
Weak Segments Performance#
This notebook provides an overview for using and understanding the weak segment performance check.
Structure:
What is the purpose of the check?#
The check is designed to easily identify the model’s weakest segments. The segments are characterized by the image properties such as contrast and aspect ratio.
Automatically detecting weak segments#
The check performs several steps:
We calculate the image properties for each sample. The properties to calculate can be passed explicitly or resort to the default image properties.
We calculate loss for each sample in the dataset using the provided model or predictions, the loss function can be passed explicitly or set to a default based on the task type.
We train multiple simple tree based models, each one is trained using two properties to predict the per sample error calculated before.
We convert each of the leafs in each of the trees into a segment and calculate the segment’s performance. For the weakest segments detected we also calculate the model’s performance on the data segments surrounding them.
Generate Dataset#
Note
In this example, we use the pytorch version of the coco dataset and model. In order to run this example using tensorflow, please change the import statements to:
from deepchecks.vision.datasets.detection import coco_tensorflow as coco
from deepchecks.vision.checks import WeakSegmentsPerformance
from deepchecks.vision.datasets.detection import coco_torch as coco
coco_data = coco.load_dataset(train=False, object_type='VisionData')
Run the check#
check = WeakSegmentsPerformance()
result = check.run(coco_data)
result
Processing Batches:
| | 0/1 [Time: 00:00]
Processing Batches:
|#####| 1/1 [Time: 00:00]
Processing Batches:
|#####| 1/1 [Time: 00:00]
Computing Check:
| | 0/1 [Time: 00:00]
Computing Check:
|#####| 1/1 [Time: 00:05]
Computing Check:
|#####| 1/1 [Time: 00:05]
To display the results in an IDE like PyCharm, you can use the following code:
# result.show_in_window()
The result will be displayed in a new window.
Observe the check’s output#
We see in the results that the check indeed found several segments on which the model performance is below average. In the heatmap display we can see the model’s performance on the weakest segments and their environment with respect to the two segmentation features. In order to get the full list of weak segments found we can look at the result.value attribute.
result.value['weak_segments_list']
Now we will run a check with properties and minimum segment size ratio (the minimal fraction of the data to be considered as a segment) different from the defaults.
from deepchecks.vision.utils.image_properties import brightness, texture_level
properties = [{'name': 'brightness', 'method': brightness, 'output_type': 'numerical'},
{'name': ' texture', 'method': texture_level, 'output_type': 'numerical'}]
check = WeakSegmentsPerformance(segment_minimum_size_ratio=0.03, image_properties=properties)
result = check.run(coco_data)
result.show()
Processing Batches:
| | 0/1 [Time: 00:00]
Processing Batches:
|#####| 1/1 [Time: 00:00]
Processing Batches:
|#####| 1/1 [Time: 00:00]
Computing Check:
| | 0/1 [Time: 00:00]
Computing Check:
|#####| 1/1 [Time: 00:00]
Computing Check:
|#####| 1/1 [Time: 00:00]
Define a condition#
We can define on our check a condition that will validate that the ratio of the model performance on the weakest segment to the average model performance is less than a specified ratio.
Processing Batches:
| | 0/1 [Time: 00:00]
Processing Batches:
|#####| 1/1 [Time: 00:00]
Processing Batches:
|#####| 1/1 [Time: 00:00]
Computing Check:
| | 0/1 [Time: 00:00]
Computing Check:
|#####| 1/1 [Time: 00:00]
Computing Check:
|#####| 1/1 [Time: 00:00]
Total running time of the script: ( 0 minutes 8.359 seconds)