Bläddra i källkod

update setup.py

FlyingQianMM 4 år sedan
förälder
incheckning
fd08f51db8
3 ändrade filer med 4 tillägg och 4 borttagningar
  1. 1 1
      dygraph/README.md
  2. 1 1
      dygraph/paddlex/utils/checkpoint.py
  3. 2 2
      dygraph/setup.py

+ 1 - 1
dygraph/README.md

@@ -66,7 +66,7 @@
       * [模型训练](tutorials/train)
       * [训练参数调整](./docs//parameters.md)
     * [VisualDL可视化训练指标](../docs/train/visualdl.md)
-    * [加载训好的模型完成预测及预测结果可视化](../docs/train/prediction.md)
+    * [加载训好的模型完成预测及预测结果可视化](./docs/apis/prediction.md)
 
 * **PaddleX GUI开发模式:**
 

+ 1 - 1
dygraph/paddlex/utils/checkpoint.py

@@ -148,7 +148,7 @@ imagenet_weights = {
     'https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/MobileNetV3_large_x1_0_pretrained.pdparams',
     'MobileNetV3_large_x1_25_IMAGENET':
     'https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/MobileNetV3_large_x1_25_pretrained.pdparams',
-    'MobileNetV3_large_x1_0_ssld':
+    'MobileNetV3_large_x1_0_ssld_IMAGENET':
     'https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/MobileNetV3_large_x1_0_ssld_pretrained.pdparams',
     'AlexNet_IMAGENET':
     'https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/AlexNet_pretrained.pdparams',

+ 2 - 2
dygraph/setup.py

@@ -19,7 +19,7 @@ long_description = "PaddlePaddle Entire Process Development Toolkit"
 
 setuptools.setup(
     name="paddlex",
-    version='2.0.0rc1',
+    version='2.0.0rc2',
     author="paddlex",
     author_email="paddlex@baidu.com",
     description=long_description,
@@ -31,7 +31,7 @@ setuptools.setup(
     install_requires=[
         "pycocotools;platform_system!='Windows'", 'pyyaml', 'colorama', 'tqdm',
         'paddleslim==2.1.0', 'visualdl>=2.1.1', 'shapely>=1.7.0',
-        'opencv-python', 'scipy'
+        'opencv-python', 'scipy', 'lap', 'motmetrics'
     ],
     classifiers=[
         "Programming Language :: Python :: 3",