| Metric | Value |
|---|---|
| Macro DICE | 0.7989 |
| Mean IoU | 0.6801 |
| FLOPs / image | 1.4091e+09 |
| DICE / GFLOPs | 0.5670 |
1{
2 "lr": 0.0001786162829017358,
3 "weight_decay": 0.0004384199029959522,
4 "batch_size": 16,
5 "loss_alpha": 0.45821908158215957
6}1from torchvision.models.segmentation import lraspp_mobilenet_v3_large
2import torch
3model = lraspp_mobilenet_v3_large()
4state = torch.load('pytorch_model.bin', map_location='cpu')
5model.load_state_dict(state)
6model.eval()