val.sh 415 Bytes
Newer Older
chenych's avatar
chenych committed
1
2
3
4
5
6
7
8
#!/bin/bash
echo "Testing start ..."
export HIP_VISIBLE_DEVICES=1

# resume是待测试模型地址
# coco_path是训练数据集地址,数据是coco format
# python main.py --batch_size 2 --no_aux_loss --eval --resume path/of/model --coco_path /path/to/coco
python main.py --batch_size 2 --no_aux_loss --eval --resume https://dl.fbaipublicfiles.com/detr/detr-r50-e632da11.pth --coco_path /home/datasets/COCO2017