diff --git a/docs/model_zoo.md b/docs/model_zoo.md index d41629cfcc62d314ee41470053c8b3d04dc99321..4460f7eca360cf6402b3c0c9453dc1a3b4fa9043 100644 --- a/docs/model_zoo.md +++ b/docs/model_zoo.md @@ -9,27 +9,27 @@ | 模型 | 模型大小 | 预测速度(毫秒) | Top1准确率(%) | Top5准确率(%) | | :----| :------- | :----------- | :--------- | :--------- | -| ResNet18| 46.9MB | 1.499 | 71.0 | 89.9 | -| ResNet34| 87.5MB | 2.272 | 74.6 | 92.1 | -| ResNet50| 102.7MB | 2.939 | 76.5 | 93.0 | -| ResNet101 |179.1MB | 5.314 | 77.6 | 93.6 | -| ResNet50_vd |102.8MB | 3.165 | 79.1 | 94.4 | -| ResNet101_vd| 179.2MB | 5.252 | 80.2 | 95.0 | -| ResNet50_vd_ssld |102.8MB | 3.165 | 82.4 | 96.1 | -| ResNet101_vd_ssld| 179.2MB | 5.252 | 83.7 | 96.7 | -| DarkNet53|166.9MB | 3.139 | 78.0 | 94.1 | -| MobileNetV1 | 16.0MB | 32.523 | 71.0 | 89.7 | -| MobileNetV2 | 14.0MB | 23.318 | 72.2 | 90.7 | -| MobileNetV3_large| 21.0MB | 19.308 | 75.3 | 93.2 | -| MobileNetV3_small | 12.0MB | 6.546 | 68.2 | 88.1 | -| MobileNetV3_large_ssld| 21.0MB | 19.308 | 79.0 | 94.5 | -| MobileNetV3_small_ssld | 12.0MB | 6.546 | 71.3 | 90.1 | -| Xception41 |92.4MB | 4.408 | 79.6 | 94.4 | -| Xception65 | 144.6MB | 6.464 | 80.3 | 94.5 | -| DenseNet121 | 32.8MB | 4.371 | 75.7 | 92.6 | -| DenseNet161|116.3MB | 8.863 | 78.6 | 94.1 | -| DenseNet201| 84.6MB | 8.173 | 77.6 | 93.7 | -| ShuffleNetV2 | 9.0MB | 10.941 | 68.8 | 88.5 | +| ResNet18| 46.9MB | 1.3568 | 71.0 | 89.9 | +| ResNet34| 87.5MB | 2.23092 | 74.6 | 92.1 | +| ResNet50| 102.7MB | 2.63824 | 76.5 | 93.0 | +| ResNet101 |179.1MB | 5.04037 | 77.6 | 93.6 | +| ResNet50_vd |102.8MB | 2.65164 | 79.1 | 94.4 | +| ResNet101_vd| 179.2MB | 5.05972 | 80.2 | 95.0 | +| ResNet50_vd_ssld |102.8MB | 2.65164 | 82.4 | 96.1 | +| ResNet101_vd_ssld| 179.2MB | 5.05972 | 83.7 | 96.7 | +| DarkNet53|166.9MB | 3.18101 | 78.0 | 94.1 | +| MobileNetV1 | 16.0MB | 0.71942 | 71.0 | 89.7 | +| MobileNetV2 | 14.0MB | 2.78603 | 72.2 | 90.7 | +| MobileNetV3_large| 21.0MB | 2.20149 | 75.3 | 93.2 | +| MobileNetV3_small | 12.0MB | 1.73933 | 68.2 | 88.1 | +| MobileNetV3_large_ssld| 21.0MB | 2.20149 | 79.0 | 94.5 | +| MobileNetV3_small_ssld | 12.0MB | 1.73933 | 71.3 | 90.1 | +| Xception41 |92.4MB | 2.85934 | 79.6 | 94.4 | +| Xception65 | 144.6MB | 4.06803 | 80.3 | 94.5 | +| DenseNet121 | 32.8MB | 4.16436 | 75.7 | 92.6 | +| DenseNet161|116.3MB | 9.27249 | 78.6 | 94.1 | +| DenseNet201| 84.6MB | 7.9617 | 77.6 | 93.7 | +| ShuffleNetV2 | 9.0MB | 1.95064 | 68.8 | 88.5 | ## 目标检测模型 diff --git a/paddlex/cv/transforms/cls_transforms.py b/paddlex/cv/transforms/cls_transforms.py index 6773f2ec469adcca673ef55e80d3a36a24d51b83..499baf7a1f4b7a8fad2ad963dee32781d71da709 100644 --- a/paddlex/cv/transforms/cls_transforms.py +++ b/paddlex/cv/transforms/cls_transforms.py @@ -87,9 +87,9 @@ class Compose(ClsTransform): import imgaug.augmenters as iaa if isinstance(op, iaa.Augmenter): im, = execute_imgaug(op, im) - output = (im, ) + outputs = (im, ) if label is not None: - output = (im, label) + outputs = (im, label) return outputs