#!/bin/bash
# 这需要在linux中进行
root_dir="/workspace/lxmodels2"
image_src_dir="${root_dir}/images"

echo "root_dir=${root_dir}"
echo "image_src_dir=${image_src_dir}"

#fp32
python3 -m bmnetp --mode="compile" \
      --model=best.torchscript.cpu.pt \
      --outdir="$(pwd)" \
      --target=BM1684 \
      --shapes=[[1,3,640,640]] \
      --net_name="best" \
      --opt=1 \
      --dyn=False \
      --cmp=True \
      --enable_profile=True 
mv compilation.bmodel best_fp32_1b.bmodel

#int8
python3 -m ufw.cali.cali_model  \
    --net_name="lxdc_lightsmoke"  \
    --model="best.torchscript.cpu.pt"  \
    --cali_image_path=${image_src_dir}  \
    --cali_iterations=10   \
    --cali_image_preprocess='resize_h=640,resize_w=640;scale=0.003921569,bgr2rgb=True'   \
    --input_shapes="[1,3,640,640]"  \
    --outdir=${int8model_dir}   \
    --target=BM1684   \
    --convert_bmodel_cmd_opt="-opt=1"   \
    --try_cali_accuracy_opt="-fpfwd_outputs=< 24 >18,< 24 >99,< 24 >172;-th_method=ADMM"