Simon
/
yolov4-baby-yoda


  
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

	
42

	
43

	
44

	
45

	
46

	
47

	
48

	
49

	
50

	
51

	
            schema: '2.0'
stages:
  prepare-train-data:
    cmd: python prepare.py prepare baby-yoda/images baby-yoda/yolo baby-yoda/train
    deps:
    - path: baby-yoda/images
      md5: 2b2fa6d21e7b76c25cd692618cf29bfe.dir
      size: 277167526
      nfiles: 209
    - path: baby-yoda/yolo
      md5: d807369820bec9c67d91bd1ea7a736be.dir
      size: 2410
      nfiles: 25
    - path: prepare.py
      md5: 097b969ec325256c2ecd0c15bab052ce
      size: 2011
    outs:
    - path: baby-yoda/train/train.txt
      md5: c35f064552527faf2191398a4eb4535a
      size: 1789
    - path: baby-yoda/train/val.txt
      md5: 5a44cdee87c484414ace252de1ea419f
      size: 481
  train:
    cmd: python train.py -l 0.001 -pretrained ./pretrained_models/yolov4.conv.137.pth
      -classes 2 -train_label_path baby-yoda/train/train.txt -dir baby-yoda/images
    deps:
    - path: baby-yoda/images
      md5: 2b2fa6d21e7b76c25cd692618cf29bfe.dir
      size: 277167526
      nfiles: 209
    - path: baby-yoda/train/train.txt
      md5: c35f064552527faf2191398a4eb4535a
      size: 1789
    - path: baby-yoda/train/val.txt
      md5: 5a44cdee87c484414ace252de1ea419f
      size: 481
    - path: cfg.py
      md5: 07619ed6c6c0510cec81a0071e227e13
      size: 1600
    - path: pretrained_models/yolov4.conv.137.pth
      md5: 3af5a93db9b189cca49f19b30468074c
      size: 170154949
    - path: train.py
      md5: cd549c1cb2218f92acacadc00743c626
      size: 28354
    outs:
    - path: checkpoints
      md5: d751713988987e9331980363e24189ce.dir
      size: 0
      nfiles: 0