/zalo

Primary LanguagePython

👋 Hi, I’m @tiamo405

Challenge.zalo.ai

Nautilus

Tham khảo code trên colab hoặc clone và làm theo hướng dẫn ở dưới

LINK COLAB

  • B1: git clone https://github.com/tiamo405/zalo.git

  • B2. terminal

    pip install -r requirements.txt
mkdir zalo/dataset
mkdir zalo/dataset/train
mkdir zalo/dataset/test
mkdir zalo/saved_models

  • B3. Download data train, test

  • B4. unzip data

    unzip "../data_zalo-challenge.zip" -d "zalo/dataset/train"
unzip ".../label.zip" -d "zalo/dataset/train"
unzip "/content/drive/MyDrive/dataAI/zalo/test.zip"
unzip "../test2.zip" -d "zalo/dataset/test"

  • B5. Train

    python zalo/train.py --name_model resnet50 --epochs 5 --lr 0.005 --batch_size 8 --replicate 11
    • model : ['resnet50', 'mobilenet_v2', 'mobilenet_v3_small', 'mobilenet_v3_large'] ( đang update )
    • Nếu bạn muốn train model riêng hãy thay tên và ở phần parser.add_argument('--name_model'...) hãy thêm tên model vào phần choices
    • relicate : số frame crop từ 1 video, nếu muốn 1 số khác bạn cần download video về và để video tại : zalo/dataset. 
      eg: zalo/dataset/public/videos

  • B6. Test

    • data public 1 : 
      python zalo/predict.py --public public --name_model resnet50 --replicate 11
    • data public 2 : 
      python zalo/predict.py --public public2 --name_model resnet50
    • hoặc 
      bash predict.sh