From 5f75b224388a1b86b27d7b3c2d484be067ac98ae Mon Sep 17 00:00:00 2001 From: Cathy0908 <30484308+Cathy0908@users.noreply.github.com> Date: Tue, 1 Nov 2022 12:09:33 +0800 Subject: [PATCH] move test model to nfs (#222) --- data/test/face_2d_keypoints/models/epoch_400.pth | 3 --- data/test/pose/hand/hrnet_w18_256x256.pth | 3 --- data/test/pose/wholebody/models/epoch_290.pth | 3 --- .../faster_rcnn_r50_fpn_1x_coco_20200130-047c8118.pth | 3 --- ...8_coco_wholebody_384x288_dark-f5726563_20200918.pth | 3 --- .../segmentation/models/mask2former_pan_export.pth | 3 --- .../segmentation/models/mask2former_r50_instance.pth | 3 --- data/test/segmentation/models/segformer_b0.pth | 3 --- tests/predictors/test_face_keypoints_predictor.py | 3 ++- tests/predictors/test_hand_keypoints_predictor.py | 4 +++- tests/predictors/test_wholebody_keypoints_predictor.py | 8 +++++--- tests/ut_config.py | 10 ++++++++++ 12 files changed, 20 insertions(+), 29 deletions(-) delete mode 100644 data/test/face_2d_keypoints/models/epoch_400.pth delete mode 100644 data/test/pose/hand/hrnet_w18_256x256.pth delete mode 100644 data/test/pose/wholebody/models/epoch_290.pth delete mode 100644 data/test/pose/wholebody/models/faster_rcnn_r50_fpn_1x_coco_20200130-047c8118.pth delete mode 100644 data/test/pose/wholebody/models/hrnet_w48_coco_wholebody_384x288_dark-f5726563_20200918.pth delete mode 100644 data/test/segmentation/models/mask2former_pan_export.pth delete mode 100644 data/test/segmentation/models/mask2former_r50_instance.pth delete mode 100644 data/test/segmentation/models/segformer_b0.pth diff --git a/data/test/face_2d_keypoints/models/epoch_400.pth b/data/test/face_2d_keypoints/models/epoch_400.pth deleted file mode 100644 index e3e92113..00000000 --- a/data/test/face_2d_keypoints/models/epoch_400.pth +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:8298b88539874b9914b90122575880a80ca0534499e9be9953e17fc177a1c2d2 -size 3421031 diff --git a/data/test/pose/hand/hrnet_w18_256x256.pth b/data/test/pose/hand/hrnet_w18_256x256.pth deleted file mode 100644 index 3585315c..00000000 --- a/data/test/pose/hand/hrnet_w18_256x256.pth +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:8570f45c7e642288b23a1c8722ba2b9b40939f1d55c962d13c789157b16edf01 -size 117072344 diff --git a/data/test/pose/wholebody/models/epoch_290.pth b/data/test/pose/wholebody/models/epoch_290.pth deleted file mode 100644 index 38a42af6..00000000 --- a/data/test/pose/wholebody/models/epoch_290.pth +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:4f2260f781f08466ce04625f8cd14c2d8fed37f9c5c71c17dcf622c0c6e94410 -size 793200649 diff --git a/data/test/pose/wholebody/models/faster_rcnn_r50_fpn_1x_coco_20200130-047c8118.pth b/data/test/pose/wholebody/models/faster_rcnn_r50_fpn_1x_coco_20200130-047c8118.pth deleted file mode 100644 index fbdd2e26..00000000 --- a/data/test/pose/wholebody/models/faster_rcnn_r50_fpn_1x_coco_20200130-047c8118.pth +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:047c8118fc5ca88ba5ae1fab72f2cd6b070501fe3af2f3cba5cfa9a89b44b03e -size 167287506 diff --git a/data/test/pose/wholebody/models/hrnet_w48_coco_wholebody_384x288_dark-f5726563_20200918.pth b/data/test/pose/wholebody/models/hrnet_w48_coco_wholebody_384x288_dark-f5726563_20200918.pth deleted file mode 100644 index bd837301..00000000 --- a/data/test/pose/wholebody/models/hrnet_w48_coco_wholebody_384x288_dark-f5726563_20200918.pth +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:f572656370050193134ef50e27745cdcdda54986d396ff78e6169608efcdc3c1 -size 254976311 diff --git a/data/test/segmentation/models/mask2former_pan_export.pth b/data/test/segmentation/models/mask2former_pan_export.pth deleted file mode 100644 index 072472ac..00000000 --- a/data/test/segmentation/models/mask2former_pan_export.pth +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:163a344e29b965cdb6c6c24e189e84a269580d63237253f359de35e944ec5421 -size 528712836 diff --git a/data/test/segmentation/models/mask2former_r50_instance.pth b/data/test/segmentation/models/mask2former_r50_instance.pth deleted file mode 100644 index 66b0a4e9..00000000 --- a/data/test/segmentation/models/mask2former_r50_instance.pth +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:a33e6b1f5623057c6920226767c91a44a072acc27ece5ba24fdeb2a9a1bb2ba2 -size 528548036 diff --git a/data/test/segmentation/models/segformer_b0.pth b/data/test/segmentation/models/segformer_b0.pth deleted file mode 100644 index 85e68166..00000000 --- a/data/test/segmentation/models/segformer_b0.pth +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:94d7df6a4ff3c605916378304b2a00404a23d4965d226a657417061647cb46a6 -size 45361179 diff --git a/tests/predictors/test_face_keypoints_predictor.py b/tests/predictors/test_face_keypoints_predictor.py index 3f62319a..e57b9a1e 100644 --- a/tests/predictors/test_face_keypoints_predictor.py +++ b/tests/predictors/test_face_keypoints_predictor.py @@ -3,6 +3,7 @@ import unittest import cv2 +from ut_config import PRETRAINED_MODEL_FACE_2D_KEYPOINTS from easycv.predictors.face_keypoints_predictor import FaceKeypointsPredictor @@ -13,7 +14,7 @@ class FaceKeypointsPredictorWithoutDetectorTest(unittest.TestCase): print(('Testing %s.%s' % (type(self).__name__, self._testMethodName))) self.image_path = './data/test/face_2d_keypoints/data/002258.png' self.save_image_path = './data/test/face_2d_keypoints/data/result_002258.png' - self.model_path = './data/test/face_2d_keypoints/models/epoch_400.pth' + self.model_path = PRETRAINED_MODEL_FACE_2D_KEYPOINTS self.model_config_path = './configs/face/face_96x96_wingloss.py' def test_single(self): diff --git a/tests/predictors/test_hand_keypoints_predictor.py b/tests/predictors/test_hand_keypoints_predictor.py index 4a325098..619a1884 100644 --- a/tests/predictors/test_hand_keypoints_predictor.py +++ b/tests/predictors/test_hand_keypoints_predictor.py @@ -2,6 +2,8 @@ import unittest +from ut_config import PRETRAINED_MODEL_HAND_KEYPOINTS + from easycv.predictors.hand_keypoints_predictor import HandKeypointsPredictor from easycv.utils.config_tools import mmcv_config_fromfile @@ -16,7 +18,7 @@ class HandKeypointsPredictorTest(unittest.TestCase): print(('Testing %s.%s' % (type(self).__name__, self._testMethodName))) self.image_path = 'data/test/pose/hand/data/hand.jpg' self.save_image_path = 'data/test/pose/hand/data/hand_result.jpg' - self.model_path = 'data/test/pose/hand/hrnet_w18_256x256.pth' + self.model_path = PRETRAINED_MODEL_HAND_KEYPOINTS self.model_config_path = 'configs/pose/hand/hrnet_w18_coco_wholebody_hand_256x256_dark.py' def test_single(self): diff --git a/tests/predictors/test_wholebody_keypoints_predictor.py b/tests/predictors/test_wholebody_keypoints_predictor.py index 623cdaea..5053b1ca 100644 --- a/tests/predictors/test_wholebody_keypoints_predictor.py +++ b/tests/predictors/test_wholebody_keypoints_predictor.py @@ -2,11 +2,13 @@ import unittest +from ut_config import (PRETRAINED_MODEL_WHOLEBODY, + PRETRAINED_MODEL_WHOLEBODY_DETECTION) + from easycv.predictors.wholebody_keypoints_predictor import \ WholeBodyKeypointsPredictor -from easycv.utils.config_tools import mmcv_config_fromfile -DEFAULT_WHOLEBODY_DETECTION_MODEL_PATH = 'data/test/pose/wholebody/models/epoch_290.pth' +DEFAULT_WHOLEBODY_DETECTION_MODEL_PATH = PRETRAINED_MODEL_WHOLEBODY_DETECTION DEFAULT_WHOLEBODY_DETECTION_CONFIG_FILE = 'configs/detection/yolox/yolox_x_8xb8_300e_coco.py' @@ -16,7 +18,7 @@ class WholeBodyKeypointsPredictorTest(unittest.TestCase): print(('Testing %s.%s' % (type(self).__name__, self._testMethodName))) self.image_path = 'data/test/pose/wholebody/data/img_test_wholebody.jpg' self.save_image_path = 'img_test_wholebody_ret.jpg' - self.model_path = 'data/test/pose/wholebody/models/hrnet_w48_coco_wholebody_384x288_dark-f5726563_20200918.pth' + self.model_path = PRETRAINED_MODEL_WHOLEBODY self.model_config_path = 'configs/pose/wholebody/hrnet_w48_coco_wholebody_384x288_dark_plus.py' def test_single(self): diff --git a/tests/ut_config.py b/tests/ut_config.py index 96620cb0..93a5cf5e 100644 --- a/tests/ut_config.py +++ b/tests/ut_config.py @@ -153,6 +153,16 @@ PRETRAINED_MODEL_SEGFORMER = os.path.join( PRETRAINED_MODEL_BEVFORMER_BASE = os.path.join( BASE_LOCAL_PATH, 'pretrained_models/detection3d/bevformer/bevformer_base_epoch_24.pth') +PRETRAINED_MODEL_FACE_2D_KEYPOINTS = os.path.join( + BASE_LOCAL_PATH, 'pretrained_models/face_2d_keypoints/epoch_400.pth') +PRETRAINED_MODEL_HAND_KEYPOINTS = os.path.join( + BASE_LOCAL_PATH, 'pretrained_models/pose/hand/hrnet/hrnet_w18_256x256.pth') +PRETRAINED_MODEL_WHOLEBODY_DETECTION = os.path.join( + BASE_LOCAL_PATH, 'pretrained_models/pose/wholebody/epoch_290.pth') +PRETRAINED_MODEL_WHOLEBODY = os.path.join( + BASE_LOCAL_PATH, + 'pretrained_models/pose/wholebody/hrnet_w48_coco_wholebody_384x288_dark-f5726563_20200918.pth' +) MODEL_CONFIG_SEGFORMER = ( './configs/segmentation/segformer/segformer_b0_coco.py') SMALL_COCO_WHOLE_BODY_HAND_ROOT = 'data/test/pose/hand/small_whole_body_hand_coco'