bplusplus 1.2.2__py3-none-any.whl → 1.2.4__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.

This version of bplusplus might be problematic. Click here for more details.

Files changed (15) hide show
  1. bplusplus/__init__.py +13 -5
  2. bplusplus/inference.py +929 -0
  3. bplusplus/prepare.py +416 -648
  4. bplusplus/{hierarchical/test.py → test.py} +32 -9
  5. bplusplus/tracker.py +261 -0
  6. bplusplus/{hierarchical/train.py → train.py} +48 -14
  7. bplusplus-1.2.4.dist-info/METADATA +207 -0
  8. bplusplus-1.2.4.dist-info/RECORD +11 -0
  9. {bplusplus-1.2.2.dist-info → bplusplus-1.2.4.dist-info}/WHEEL +1 -1
  10. bplusplus/resnet/test.py +0 -473
  11. bplusplus/resnet/train.py +0 -329
  12. bplusplus/train_validate.py +0 -11
  13. bplusplus-1.2.2.dist-info/METADATA +0 -260
  14. bplusplus-1.2.2.dist-info/RECORD +0 -12
  15. {bplusplus-1.2.2.dist-info → bplusplus-1.2.4.dist-info}/LICENSE +0 -0
bplusplus/__init__.py CHANGED
@@ -1,7 +1,15 @@
1
+ try:
2
+ import torch
3
+ import torchvision
4
+ except ImportError:
5
+ raise ImportError(
6
+ "PyTorch and Torchvision are not installed. "
7
+ "Please install them before using bplusplus by following the instructions "
8
+ "on the official PyTorch website: https://pytorch.org/get-started/locally/"
9
+ )
10
+
1
11
  from .collect import Group, collect
2
- from .train_validate import train, validate
3
12
  from .prepare import prepare
4
- from .resnet.train import train_resnet
5
- from .resnet.test import test_resnet
6
- from .hierarchical.train import train_multitask
7
- from .hierarchical.test import test_multitask
13
+ from .train import train
14
+ from .test import test
15
+ from .inference import inference