File tree Expand file tree Collapse file tree 3 files changed +4
-4
lines changed Expand file tree Collapse file tree 3 files changed +4
-4
lines changed Original file line number Diff line number Diff line change 29
29
- pillow>=10.4.0
30
30
- pytest>=6.1.2
31
31
- scikit-image>=0.22.0
32
- - timm>=1.0.14
32
+ - timm>=1.0.15
33
33
- torch>=2.6
34
- - torchmetrics>=0.10
34
+ - torchmetrics>=1.6.2
35
35
- torchvision>=0.18
36
36
exclude : (build|data|dist|logo|logs|output)/
37
37
- repo : https://github.com/pre-commit/mirrors-prettier
Original file line number Diff line number Diff line change @@ -18,6 +18,6 @@ segmentation-models-pytorch==0.4.0
18
18
shapely==2.0.7
19
19
timm==1.0.15
20
20
torch==2.6.0
21
- torchmetrics==1.6.1
21
+ torchmetrics==1.6.2
22
22
torchvision==0.21.0
23
23
typing-extensions==4.12.2
Original file line number Diff line number Diff line change @@ -117,7 +117,7 @@ def configure_metrics(self) -> None:
117
117
* 'Macro' averaging gives equal weight to each class, and is useful for
118
118
balanced performance assessment across imbalanced classes.
119
119
"""
120
- metrics = MetricCollection ([MeanAveragePrecision (iou_type = ('bbox' , 'segm' ))]) # type: ignore[arg-type]
120
+ metrics = MetricCollection ([MeanAveragePrecision (iou_type = ('bbox' , 'segm' ))])
121
121
self .val_metrics = metrics .clone (prefix = 'val_' )
122
122
self .test_metrics = metrics .clone (prefix = 'test_' )
123
123
You can’t perform that action at this time.
0 commit comments