fix draw_net python script

include caffe.draw for drawing functions.