From c28839d24b24b741bfb1ca9343e22f4b6d0ac54c Mon Sep 17 00:00:00 2001
From: wangziyang <2890199310@qq.com>
Date: Thu, 12 May 2022 20:08:43 +0800
Subject: [PATCH] =?UTF-8?q?=E6=8F=90=E4=BA=A4=E6=BA=90=E7=A0=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/PaddleClas/tools/.train.py.un~   | Bin 0 -> 2295 bytes
 src/PaddleClas/tools/__init__.py     |  15 ++++++++++++
 src/PaddleClas/tools/eval.py         |  31 ++++++++++++++++++++++++
 src/PaddleClas/tools/eval.sh         |   7 ++++++
 src/PaddleClas/tools/export_model.py |  34 +++++++++++++++++++++++++++
 src/PaddleClas/tools/infer.py        |  31 ++++++++++++++++++++++++
 src/PaddleClas/tools/train.py        |  32 +++++++++++++++++++++++++
 src/PaddleClas/tools/train.sh        |   8 +++++++
 8 files changed, 158 insertions(+)
 create mode 100644 src/PaddleClas/tools/.train.py.un~
 create mode 100644 src/PaddleClas/tools/__init__.py
 create mode 100644 src/PaddleClas/tools/eval.py
 create mode 100644 src/PaddleClas/tools/eval.sh
 create mode 100644 src/PaddleClas/tools/export_model.py
 create mode 100644 src/PaddleClas/tools/infer.py
 create mode 100644 src/PaddleClas/tools/train.py
 create mode 100644 src/PaddleClas/tools/train.sh

diff --git a/src/PaddleClas/tools/.train.py.un~ b/src/PaddleClas/tools/.train.py.un~
new file mode 100644
index 0000000000000000000000000000000000000000..1b2c2905a05c1eb0ed871fbb29bd78dbd980d823
GIT binary patch
literal 2295
zcmWH`%$*;a=aT=FfobwCrVkHZ9qkES@v0%uRpz@+;+c)iA+MMGd+oxsqNFODfq_8@
zh(SOSLNPD`X;vs3M8hycQp_na7LY76L;MA(0wyR8qCir7F#dla0E@r`MrkUCKvK*}
zS#WePGJKT)n#~Ku{6MS##Gt5PFl_`!6C}n!0uTU-GEg*$fw-g51PTRO0$?=F0!I@A
zB${}DmcgP46cq}}3OSj%nI(xOnfZCe3Z;1|sYMDU8L0|BnaQbn#i@EA$6<|Nv4X^u
zl$=z()V#!`oYeSYkP*omn$Tzmnau*y3kA$T8kWUJqa7A1G^2shz6?3qK{-?!h!ubs
zId_7h1y<SuncyfJjV4e?&;<aaX$3f%m>_u!7EOvk+mWLQIVXV}28y!LXaa=<T>!A@
WAqtKrW<)ee0c}<SVrVoyUj+c1I$cr#

literal 0
HcmV?d00001

diff --git a/src/PaddleClas/tools/__init__.py b/src/PaddleClas/tools/__init__.py
new file mode 100644
index 0000000..cdefb80
--- /dev/null
+++ b/src/PaddleClas/tools/__init__.py
@@ -0,0 +1,15 @@
+# Copyright (c) 2020 PaddlePaddle Authors. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from . import infer
\ No newline at end of file
diff --git a/src/PaddleClas/tools/eval.py b/src/PaddleClas/tools/eval.py
new file mode 100644
index 0000000..e086da1
--- /dev/null
+++ b/src/PaddleClas/tools/eval.py
@@ -0,0 +1,31 @@
+# Copyright (c) 2021 PaddlePaddle Authors. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from __future__ import absolute_import
+from __future__ import division
+from __future__ import print_function
+import os
+import sys
+__dir__ = os.path.dirname(os.path.abspath(__file__))
+sys.path.append(os.path.abspath(os.path.join(__dir__, '../')))
+
+from ppcls.utils import config
+from ppcls.engine.engine import Engine
+
+if __name__ == "__main__":
+    args = config.parse_args()
+    config = config.get_config(
+        args.config, overrides=args.override, show=False)
+    engine = Engine(config, mode="eval")
+    engine.eval()
diff --git a/src/PaddleClas/tools/eval.sh b/src/PaddleClas/tools/eval.sh
new file mode 100644
index 0000000..c13ea6d
--- /dev/null
+++ b/src/PaddleClas/tools/eval.sh
@@ -0,0 +1,7 @@
+#!/usr/bin/env bash
+
+# for single card eval
+# python3.7 tools/eval.py -c ./ppcls/configs/ImageNet/ResNet/ResNet50.yaml
+
+# for multi-cards eval
+python3.7 -m paddle.distributed.launch --gpus="0,1,2,3" tools/eval.py -c ./ppcls/configs/ImageNet/ResNet/ResNet50.yaml
diff --git a/src/PaddleClas/tools/export_model.py b/src/PaddleClas/tools/export_model.py
new file mode 100644
index 0000000..01aba06
--- /dev/null
+++ b/src/PaddleClas/tools/export_model.py
@@ -0,0 +1,34 @@
+# Copyright (c) 2021 PaddlePaddle Authors. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from __future__ import absolute_import
+from __future__ import division
+from __future__ import print_function
+import os
+import sys
+__dir__ = os.path.dirname(os.path.abspath(__file__))
+sys.path.append(os.path.abspath(os.path.join(__dir__, '../')))
+
+import paddle
+import paddle.nn as nn
+
+from ppcls.utils import config
+from ppcls.engine.engine import Engine
+
+if __name__ == "__main__":
+    args = config.parse_args()
+    config = config.get_config(
+        args.config, overrides=args.override, show=False)
+    engine = Engine(config, mode="export")
+    engine.export()
diff --git a/src/PaddleClas/tools/infer.py b/src/PaddleClas/tools/infer.py
new file mode 100644
index 0000000..4f6bf92
--- /dev/null
+++ b/src/PaddleClas/tools/infer.py
@@ -0,0 +1,31 @@
+# Copyright (c) 2021 PaddlePaddle Authors. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from __future__ import absolute_import
+from __future__ import division
+from __future__ import print_function
+import os
+import sys
+__dir__ = os.path.dirname(os.path.abspath(__file__))
+sys.path.append(os.path.abspath(os.path.join(__dir__, '../')))
+
+from ppcls.utils import config
+from ppcls.engine.engine import Engine
+
+if __name__ == "__main__":
+    args = config.parse_args()
+    config = config.get_config(
+        args.config, overrides=args.override, show=False)
+    engine = Engine(config, mode="infer")
+    engine.infer()
diff --git a/src/PaddleClas/tools/train.py b/src/PaddleClas/tools/train.py
new file mode 100644
index 0000000..e7c9d7b
--- /dev/null
+++ b/src/PaddleClas/tools/train.py
@@ -0,0 +1,32 @@
+# Copyright (c) 2021 PaddlePaddle Authors. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from __future__ import absolute_import
+from __future__ import division
+from __future__ import print_function
+import os
+import sys
+__dir__ = os.path.dirname(os.path.abspath(__file__))
+sys.path.append(os.path.abspath(os.path.join(__dir__, '../')))
+
+from ppcls.utils import config
+from ppcls.engine.engine import Engine
+
+if __name__ == "__main__":
+    args = config.parse_args()
+    config = config.get_config(
+        args.config, overrides=args.override, show=False)
+    config.profiler_options = args.profiler_options
+    engine = Engine(config, mode="train")
+    engine.train()
diff --git a/src/PaddleClas/tools/train.sh b/src/PaddleClas/tools/train.sh
new file mode 100644
index 0000000..412f108
--- /dev/null
+++ b/src/PaddleClas/tools/train.sh
@@ -0,0 +1,8 @@
+#!/usr/bin/env bash
+
+# for single card train
+# python3.7 tools/train.py -c ./ppcls/configs/ImageNet/ResNet/ResNet50.yaml
+
+# for multi-cards train
+export CUDA_VISIBLE_DEVICES=0,1,2,3
+python3.7 -m paddle.distributed.launch --gpus="0,1,2,3" tools/train.py -c ./ppcls/configs/ImageNet/ResNet/ResNet50.yaml