Fix PyPI package

README.rst

@@ -61,8 +61,6 @@ The application can also be built into a Docker image and hosted on Docker Hub.
 
 .. code-block:: console
 
-    # linux/amd64
-
     docker login
 
     export DOCKER_BUILDKIT=1
@@ -71,7 +69,7 @@ The application can also be built into a Docker image and hosted on Docker Hub.
 
     docker buildx build \
         -t wildme/scoutbot:latest \
-        --platform linux/arm64 \
+        --platform linux/amd64 \
         --push \
         .
 

scoutbot/__init__.py

@@ -37,7 +37,7 @@ Notes:
 '''
 from scoutbot import agg, loc, tile, wic
 
-VERSION = '0.1.1'
+VERSION = '0.1.2'
 version = VERSION
 __version__ = VERSION
 

scoutbot/loc/transforms/_postprocess.py

@@ -10,8 +10,6 @@ import logging
 import torch
 
 # from torch.autograd import Variable
-from scoutbot.loc.transforms.detections.detection import Detection
-
 from .util import BaseTransform
 
 __all__ = [
@@ -234,6 +232,8 @@ class TensorToBrambox(BaseTransform):
 
     @staticmethod
     def _convert(boxes, width, height, class_label_map):
+        from scoutbot.loc.transforms.detections.detection import Detection
+
         boxes[:, 0:3:2].mul_(width)
         boxes[:, 0] -= boxes[:, 2] / 2
         boxes[:, 1:4:2].mul_(height)

setup.cfg

@@ -18,19 +18,20 @@ packages = find:
 platforms = any
 include_package_data = True
 install_requires =
-    onnxruntime
+    click
+    cryptography
+    gradio
+    imgaug
     numpy
-    wbia-utool
-    torch
-    torchvision
+    onnxruntime-gpu
     opencv-python-headless
     Pillow
-    imgaug
+    pooch
     rich
+    torch
+    torchvision
     tqdm
-    gradio
-    cryptography
-    click
+    wbia-utool
 python_requires = >=3.7
 
 [options.entry_points]