Browse Source

Fix pip installation 2

boomb0om 2 years ago
parent
commit
053ca05ff8
1 changed files with 1 additions and 0 deletions
  1. 1 0
      CRAFT/__init__.py

+ 1 - 0
CRAFT/__init__.py

@@ -1,2 +1,3 @@
 from .model import CRAFTModel, preprocess_image
 from .utils import draw_boxes, draw_polygons, boxes_area, polygons_area
+from .basenet import vgg16_bn, init_weights