|
a |
|
b/.github/workflows/EfficientNetB4.yml |
|
|
1 |
name: EfficientNetB4 |
|
|
2 |
|
|
|
3 |
on: |
|
|
4 |
push: |
|
|
5 |
branches: [master, dev] |
|
|
6 |
|
|
|
7 |
jobs: |
|
|
8 |
Train: |
|
|
9 |
runs-on: ubuntu-18.04 |
|
|
10 |
steps: |
|
|
11 |
- uses: actions/checkout@v2 |
|
|
12 |
- name: Set up Python 3.7 |
|
|
13 |
uses: actions/setup-python@v1 |
|
|
14 |
with: |
|
|
15 |
python-version: 3.7 |
|
|
16 |
- name: Install dependencies |
|
|
17 |
run: | |
|
|
18 |
python -m pip install -r requirements.txt |
|
|
19 |
- name: Generate data files |
|
|
20 |
run: | |
|
|
21 |
python dataset-generation-pool.py |
|
|
22 |
working-directory: ./scripts |
|
|
23 |
- name: Generate 2D annotation files |
|
|
24 |
run: | |
|
|
25 |
python annotation-generation-2d.py |
|
|
26 |
working-directory: ./scripts |
|
|
27 |
- name: Running training |
|
|
28 |
run: | |
|
|
29 |
python train.py --config configs/training/EfficientNetB4.json |