Git Product home page Git Product logo

zalo_traffic_detection's Introduction

zalo_traffic_detection

Prepare data

cd /model/prepare_data

download data

wget --header="Host: dl.challenge.zalo.ai" --header="User-Agent: Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/86.0.4240.111 Safari/537.36" --header="Accept: text/html,application/xhtml+xml,application/xml;q=0.9,image/avif,image/webp,image/apng,*/*;q=0.8,application/signed-exchange;v=b3;q=0.9" --header="Accept-Language: en-US,en;q=0.9,vi-VN;q=0.8,vi;q=0.7" --header="Referer: https://challenge.zalo.ai/" --header="Cookie: _ga=GA1.2.1869849351.1604392068; _gid=GA1.2.2084818595.1604392068; __zi=3000.SSZzejyD0jydXQcYsa00d3xBfxgP71AM8Tdbg8y9KSbdtERbmGzGp6FRgU7211d5RzAaxCmD6Sa.1; zaiac_c=6elQ0Vlhi3COWzu6_UIqE1w3t0_YkDHWMe-x3E6HWtnNgECSnB_CSYMgnmN2YyXpPAd64OMMvYKviOmHejBdOZZt_pEIvV1M8lF08Rdyj5SPz88nai-KP0Bd-G2Qz90m9_EAGy2MaNLsygCTmiUGLLJF-XRawCbJRDhaOyh0Y2LjvAPhpVlELIdUqYEL_-Pl9js899RUg4ycz9G1ZkZvDpdK_qUD_Tuc1fZp5-Qb-NHJakW2j97BI0gatGwUmlSu2yZlSP-XtoOBXCnjkhYa05FIK5a7-_MoDG; fpsend=148555; _gat_gtag_UA_108352130_23=1" --header="Connection: keep-alive" "https://dl.challenge.zalo.ai/traffic-sign-detection/data/za_traffic_2020.zip" -c -O 'za_traffic_2020.zip'

unzip za_traffic_2020.zip

merge overlay bbox

python3 merge_over_lay_bbox.py

k fold

python3 k_fold.py

split big image to small image

python3 split_image.py

create data fold

python3 create_data_fold.py --fold <1 to 5>
python3 create_label_fold.py --fold <1 to 5>

train

cd /model/train

train

python3 train.py --img 160 --batch <sô lưởng ảnh trong 1 batch size> --epochs 50 --data zalo_traffic_160_fold<1 to 5>.yaml --weights '' --cfg yolov5x_zalo_traffic.yaml --hyp data/hyp.zalo.yaml --multi-scale --worker 20 

ví dụ train với 2 card 2080ti fold1

python3 train.py --img 160 --batch 160 --epochs 50 --data zalo_traffic_160_fold1.yaml --weights '' --cfg yolov5x_zalo_traffic.yaml --hyp data/hyp.zalo.yaml --multi-scale --worker 20 

log

log và model được lưu trong thư mục run/exp*

zalo_traffic_detection's People

Contributors

freedom1810 avatar thuyduongbka avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.