Fix installation

CRAFT/__init__.py

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

CRAFT/craft.py

@@ -8,9 +8,9 @@ import torch
 import torch.nn as nn
 import torch.nn.functional as F
 
-from .basenet import vgg16_bn, init_weights
-from .utils import copyStateDict
-from .fp16 import FP16Module
+from CRAFT.basenet import vgg16_bn, init_weights
+from CRAFT.utils import copyStateDict
+from CRAFT.fp16 import FP16Module
 
 
 class double_conv(nn.Module):

CRAFT/model.py

@@ -8,10 +8,10 @@ import numpy as np
 import cv2
 from huggingface_hub import hf_hub_url, cached_download
 
-from .craft import CRAFT, init_CRAFT_model
-from .refinenet import RefineNet, init_refiner_model
-from .craft_utils import adjustResultCoordinates, getDetBoxes
-from .imgproc import resize_aspect_ratio, normalizeMeanVariance
+from CRAFT.craft import CRAFT, init_CRAFT_model
+from CRAFT.refinenet import RefineNet, init_refiner_model
+from CRAFT.craft_utils import adjustResultCoordinates, getDetBoxes
+from CRAFT.imgproc import resize_aspect_ratio, normalizeMeanVariance
 
 
 HF_MODELS = {

setup.py

@@ -10,7 +10,7 @@ setup(
     description="",
     author="Igor Pavlov, Youngmin Baek, Bado Lee, Dongyoon Han, Sangdoo Yun, Hwalsuk Lee",
     url='https://github.com/boomb0om/CRAFT-text-detection',
-    packages=find_packages(include=['CRAFT', 'CRAFT/basenet']),
+    packages=find_packages(include=['CRAFT', 'CRAFT.*']),
     install_requires=[
         str(r)
         for r in pkg_resources.parse_requirements(