https://github.com/um-arm-lab/siammask
https://github.com/um-arm-lab/siammask
Last synced: 11 months ago
JSON representation
- Host: GitHub
- URL: https://github.com/um-arm-lab/siammask
- Owner: UM-ARM-Lab
- Created: 2019-09-10T20:29:06.000Z (over 6 years ago)
- Default Branch: master
- Last Pushed: 2020-06-06T07:11:43.000Z (almost 6 years ago)
- Last Synced: 2025-02-22T17:47:28.502Z (over 1 year ago)
- Language: Python
- Size: 17.5 MB
- Stars: 1
- Watchers: 12
- Forks: 0
- Open Issues: 1
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
# SiamMask ROS Version
## Run Message subscriber
```
rosrun SiamMask run_demo --resume SiamMask_DAVIS.pth --config config_davis.json
rostopic pub /SiamMask_BBox mps_msgs/AABBox2d "xmin: 50.0
ymin: 100.0
xmax: 100.0
ymax: 200.0"
```
## Run Action Server
```
rosrun SiamMask siamserver --resume SiamMask_DAVIS.pth --config config_davis.json
```
## Environment
```
export PYTHONPATH=/home/kunhuang/catkin_ws/devel/lib/python3/dist-packages:/home/kunhuang/.pyenv/versions/anaconda2-2019.03/envs/siammask/lib/python3.6/site-packages/cv2:/home/kunhuang/local/lib/python2.7/dist-packages:/home/kunhuang/mps_ws/devel/lib/python2.7/dist-packages:/home/kunhuang/armlab_ws/devel/lib/python2.7/dist-packages:/opt/ros/melodic/lib/python2.7/dist-packages
export ROS_PACKAGE_PATH=/home/kunhuang/mps_ws/src:/home/kunhuang/armlab_ws/src:/home/kunhuang/catkin_ws/src:/opt/ros/melodic/share
```
# DOCKER RELATED
## Build Images
Baseline (CUDA, ROS, other common):
```
docker build . -f ci/Dockerfile_baseline -t um_arm_lab/baseline:melodic
```
SiamMask:
```
docker build . -f ci/Dockerfile -t um_arm_lab/siam_mask:melodic
```
## Run demo.py
```
docker run --gpus all -it -v /tmp/.X11-unix:/tmp/.X11-unix -e DISPLAY=unix$DISPLAY um_arm_lab/siam_mask:melodic
```
```
source /root/catkin_ws/devel/setup.bash
export SIAM_MASK_PATH=/root/catkin_ws/src/siam_mask/src/SiamMask/
export PYTHONPATH=/root/catkin_ws/devel/lib/python2.7/dist-packages:/opt/ros/melodic/lib/python2.7/dist-packages:$SIAM_MASK_PATH/experiments/siammask_sharp:$SIAM_MASK_PATH
cd $SIAM_MASK_PATH/experiments/siammask_sharp
python ../../tools/demo.py --resume SiamMask_DAVIS.pth --config config_davis.json
```