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

	
            stages:
  prepare-train-data:
    cmd: python prepare.py prepare baby-yoda/images baby-yoda/yolo baby-yoda/train
    deps:
    - baby-yoda/images
    - baby-yoda/yolo
    - prepare.py
    outs:
    - baby-yoda/train/train.txt
    - baby-yoda/train/val.txt
  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:
    - baby-yoda/images
    - baby-yoda/train/train.txt
    - baby-yoda/train/val.txt
    - cfg.py
    - pretrained_models/yolov4.conv.137.pth
    - train.py
    outs:
    - checkpoints