python tools/train_net.py
–config-file configs/FCOS-Detection/R_50_1x.yaml
–eval-only
–num-gpus 2
OUTPUT_DIR training_dir/fcos_R_50_1x
MODEL.WEIGHTS training_dir/fcos_R_50_1x/fcos_R_50_1x.pth
File “/home/xxx/AdelaiDet/adet/layers/bezier_align.py”, line 7, in
from adet import _C
ImportError: adet/_C.cpython-37m-x86_64-linux-gnu.so: undefined symbol: _ZNK2at6Tensor6deviceEv
在build之前需要先检查一下。
git clone https://github.com/facebookresearch/detectron2.git
cd detectron2
git checkout -f 9eb4831
cd ..
python -m pip install -e detectron2
git clone https://github.com/aim-uofa/AdelaiDet.git
cd AdelaiDet
python setup.py build develop
此外,需要注意,build之前,必须删除旧的构建文件
rm -rf build