-
Notifications
You must be signed in to change notification settings - Fork 120
/
init.sh
31 lines (20 loc) · 913 Bytes
/
init.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
# Download pretrained models
if [ ! -f model/pretrained_model/resnet-101-caffe.pth ]; then
curl http://www.yuwenxiong.com/pretrained_model/resnet-101-caffe.pth -o model/pretrained_model/resnet-101-caffe.pth
fi
if [ ! -f model/pretrained_model/resnet-50-caffe.pth ]; then
curl http://www.yuwenxiong.com/pretrained_model/resnet-50-caffe.pth -o model/pretrained_model/resnet-50-caffe.pth
fi
# Install essential python packages
pip install pyyaml pycocotools
# Download panopticapi devkit
git clone https://github.com/cocodataset/panopticapi lib/dataset_devkit/panopticapi
# Build essential operators
# build cython modules
cd upsnet/bbox; python setup.py build_ext --inplace
cd ../rpn; python setup.py build_ext --inplace
cd ../nms; python setup.py build_ext --inplace
# build operators
cd ../operators
python build_deform_conv.py build_ext --inplace
python build_roialign.py build_ext --inplace