Commit dfd658ff authored by chenych's avatar chenych
Browse files

Add .sh

parent b952e97b
*.mdb
*.pth
*.tar
*.sh
*.txt
*.ipynb
*.zip
......
......@@ -127,7 +127,7 @@ WIDER_FACE验证集上的测试结果如下
./models/model_best.pth
## 源码仓库及问题反馈
https://developer.hpccube.com/codes/modelzoo/centerface-pytorch
https://developer.hpccube.com/codes/modelzoo/centerface_pytorch
## 参考资料
https://github.com/chenjun2hao/CenterFace.pytorch
#!/usr/bin/env bash
python3 setup.py build_ext --inplace
rm -rf build
\ No newline at end of file
python setup.py build_ext --inplace
#!/usr/bin/env bash
python main.py --input_h 800 --input_w 800 --batch_size 16 --lr 6.25e-5 --load_model /path/of/model_best.pth --data_dir /your/WIDER_train/dir/path
\ No newline at end of file
#!/usr/bin/env bash
python main.py --input_h 800 --input_w 800 --batch_size 16 --lr 6.25e-5 --gpus 0,1 --load_model /path/of/model_best.pth --data_dir /your/WIDER_train/dir/path
\ No newline at end of file
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment