Explorar o código

update cls det seg

will-jl944 %!s(int64=4) %!d(string=hai) anos
pai
achega
2abb97f979
Modificáronse 3 ficheiros con 10 adicións e 19 borrados
  1. 1 6
      dygraph/paddlex/cls.py
  2. 1 6
      dygraph/paddlex/det.py
  3. 8 7
      dygraph/paddlex/seg.py

+ 1 - 6
dygraph/paddlex/cls.py

@@ -14,15 +14,10 @@
 
 from paddlex.cv.transforms import cls_transforms
 
-message = 'Your running script needs PaddleX<2.0.0, please refer to {} to solve this issue.'.format(
-    'https://github.com/PaddlePaddle/PaddleX/tree/release/2.0-rc/tutorials/train#%E7%89%88%E6%9C%AC%E5%8D%87%E7%BA%A7'
-)
-
 
 def __getattr__(attr):
     if attr == 'transforms':
-        print("\033[1;31;40m{}\033[0m".format(message).encode("utf-8")
-              .decode("latin1"))
+        print("You are using PaddleX 1.x API")
 
 
 transforms = cls_transforms

+ 1 - 6
dygraph/paddlex/det.py

@@ -14,15 +14,10 @@
 
 from paddlex.cv.transforms import det_transforms
 
-message = 'Your running script needs PaddleX<2.0.0, please refer to {} to solve this issue.'.format(
-    'https://github.com/PaddlePaddle/PaddleX/tree/release/2.0-rc/tutorials/train#%E7%89%88%E6%9C%AC%E5%8D%87%E7%BA%A7'
-)
-
 
 def __getattr__(attr):
     if attr == 'transforms':
-        print("\033[1;31;40m{}\033[0m".format(message).encode("utf-8")
-              .decode("latin1"))
+        print("You are using PaddleX 1.x API")
 
 
 transforms = det_transforms

+ 8 - 7
dygraph/paddlex/seg.py

@@ -11,18 +11,19 @@
 # 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.
-
+import paddlex.cv.models
 from paddlex.cv.transforms import seg_transforms
 
-message = 'Your running script needs PaddleX<2.0.0, please refer to {} to solve this issue.'.format(
-    'https://github.com/PaddlePaddle/PaddleX/tree/release/2.0-rc/tutorials/train#%E7%89%88%E6%9C%AC%E5%8D%87%E7%BA%A7'
-)
-
 
 def __getattr__(attr):
     if attr == 'transforms':
-        print("\033[1;31;40m{}\033[0m".format(message).encode("utf-8")
-              .decode("latin1"))
+        print("You are using PaddleX 1.x API")
 
 
 transforms = seg_transforms
+
+UNet = paddlex.cv.models.UNet
+DeepLabV3P = paddlex.cv.models.DeepLabV3P
+FastSCNN = paddlex.cv.models.FastSCNN
+HRNet = paddlex.cv.models.HRNet
+BiSeNetV2 = paddlex.cv.models.BiSeNetV2