#875 Feature/sg 761 yolo nas

Merged

Ghost merged 1 commits into Deci-AI:master from deci-ai:feature/SG-761-yolo-nas

  
1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

            defaults:
  - cityscapes_dataset_params
  - _self_
train_dataset_params:
  transforms:
    - SegColorJitter:
        brightness: 0.5
        contrast: 0.5
        saturation: 0.5
    - SegRandomFlip:
        prob: 0.5
    - SegRandomRescale:
        scales: [ 0.5, 2.0 ]
    - SegPadShortToCropSize:
        crop_size: [ 1024, 1024 ]
        fill_mask: 19
    - SegCropImageAndMask:
        crop_size: [ 1024, 1024 ]
        mode: random
val_dataset_params:
  transforms:
    - SegRescale:
        long_size: 1024
    - SegPadShortToCropSize:
        crop_size: [ 1024, 1024 ]
        fill_mask: 19
train_dataloader_params:
  batch_size: 2
  shuffle: True
val_dataloader_params:
  batch_size: 2
  shuffle: False

Tip!

Press p or to see the previous file or, n or to see the next file

Deci-AI / super-gradients connected to https://github.com/Deci-AI/super-gradients.git

#875 Feature/sg 761 yolo nas

Deci-AI
/
super-gradients
connected to https://github.com/Deci-AI/super-gradients.git