Skip to content

Commit

Permalink
Fix data augmentation default config
Browse files Browse the repository at this point in the history
  • Loading branch information
sfmig committed Jun 27, 2024
1 parent 3df8635 commit 43f852b
Showing 1 changed file with 17 additions and 4 deletions.
21 changes: 17 additions & 4 deletions crabs/detection_tracking/config/faster_rcnn.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -37,16 +37,29 @@ batch_size_test: 4
# -------------------
# Data augmentation
# -------------------
transform_brightness: 0.5
transform_hue: 0.3
gaussian_blur_params:
gaussian_blur:
kernel_size:
- 5
- 9
sigma:
- 0.1
- 5.0

color_jitter:
brightness: 0.5
hue: 0.3
random_horizontal_flip:
p: 0.5
random_rotation:
degrees: [-10.0, 10.0]
random_adjust_sharpness:
p: 0.5
sharpness_factor: 0.5
random_autocontrast:
p: 0.5
random_equalize:
p: 0.5
clamp_and_sanitize_bboxes:
min_size: 1.0
# ----------------------------
# Hyperparameter optimisation
# -----------------------------
Expand Down

0 comments on commit 43f852b

Please sign in to comment.