Skip to content

Commit

Permalink
Add Model Profile
Browse files Browse the repository at this point in the history
  • Loading branch information
ViiSkor committed Oct 31, 2023
1 parent 4da907a commit bbbfcf3
Showing 1 changed file with 61 additions and 0 deletions.
61 changes: 61 additions & 0 deletions ModelProfile.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,61 @@
# Model Profile

## Overview
- **Name:** U-Net
- **Type:** Convolutional Neural Network (CNN)
- **Date of Introduction:** 2015

## Key Characteristics

### Augmentations
```
transform = A.Compose(
[
A.HorizontalFlip(p=0.5),
A.ShiftScaleRotate(scale_limit=0.1, rotate_limit=45, shift_limit=0.0, p=0.5),
A.IAAPerspective(p=0.2),
A.OneOf(
[
A.CLAHE(p=1),
A.RandomBrightnessContrast(p=1),
A.RandomGamma(p=0.1),
],
p=0.4,
),
A.OneOf(
[
A.Sharpen (p=0.4),
A.Blur(blur_limit=3, p=0.4),
A.MotionBlur(blur_limit=3, p=0.4),
],
p=0.4,
),
A.OneOf(
[
A.RandomBrightnessContrast(p=0.7),
A.HueSaturationValue(p=0.7),
],
p=0.4,
),
])
```

### Loss Function
- 1e-3 * Binary Cross-entropy loss - Dice Loss

### Data Preprocessing
- Undersampling
- Normalization
- Del corrupted files
- Del files less than 50kb

### Training
- Training config is [here](https://github.com/ViiSkor/SatelliteImgOfShips/blob/master/config.yml)

### Final Scores
| Loss | Dice | Binary Acc | POD | Epochs |
|---------|--------|------------|--------|--------|
| -0.3859 | 0.3688 | 0.9287 | 0.6467 | 70 |

### Relevant commit for the experiment
```4da907add32589e241642892ab0560387375422a```

0 comments on commit bbbfcf3

Please sign in to comment.