Skip to content

Commit 43c4488

Browse files
shentanyueshentanyue
committed
add more model ut
1 parent f46f0a2 commit 43c4488

File tree

34 files changed

+3519
-25
lines changed

34 files changed

+3519
-25
lines changed

‎lite/CMakeLists.txt

Lines changed: 43 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -70,15 +70,15 @@ if (WITH_TESTING)
7070
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL_FOR_UNITTESTS} "SqueezeNet1_0_v2_0.tar.gz")
7171
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL_FOR_UNITTESTS} "VGG16_v2_0.tar.gz")
7272
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL_FOR_UNITTESTS} "VGG19_v2_0.tar.gz")
73-
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL_FOR_NNADAPTER_UNITTESTS} "DarkNet53.tar.gz" MODEL_PATH "PaddleClas/v2.3/DarkNet")
73+
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL_FOR_NNADAPTER_UNITTESTS} "DarkNet53.tar.gz" MODEL_PATH "PaddleClas/v2.3/DarkNet")
7474
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL_FOR_NNADAPTER_UNITTESTS} "DeiT_base_patch16_224.tar.gz" MODEL_PATH "PaddleClas/v2.3/DeiT")
75-
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL_FOR_NNADAPTER_UNITTESTS} "DPN68.tar.gz" MODEL_PATH "PaddleClas/v2.3/DPN")
76-
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL_FOR_NNADAPTER_UNITTESTS} "GhostNet_x1_0.tar.gz" MODEL_PATH "PaddleClas/v2.3/GhostNet")
77-
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL_FOR_NNADAPTER_UNITTESTS} "HRNet_W18_C.tar.gz" MODEL_PATH "PaddleClas/v2.3/HRNet")
78-
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL_FOR_NNADAPTER_UNITTESTS} "PPLCNet_x0_25.tar.gz" MODEL_PATH "PaddleClas/v2.3/PPLCNet")
79-
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL_FOR_NNADAPTER_UNITTESTS} "Res2Net50_26w_4s.tar.gz" MODEL_PATH "PaddleClas/v2.3/Res2Net")
80-
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL_FOR_NNADAPTER_UNITTESTS} "SE_ResNet50_vd.tar.gz" MODEL_PATH "PaddleClas/v2.3/SENet")
81-
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL_FOR_NNADAPTER_UNITTESTS} "ViT_base_patch16_224.tar.gz" MODEL_PATH "PaddleClas/v2.3/VisionTransformer")
75+
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL_FOR_NNADAPTER_UNITTESTS} "DPN68.tar.gz" MODEL_PATH "PaddleClas/v2.3/DPN")
76+
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL_FOR_NNADAPTER_UNITTESTS} "GhostNet_x1_0.tar.gz" MODEL_PATH "PaddleClas/v2.3/GhostNet")
77+
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL_FOR_NNADAPTER_UNITTESTS} "HRNet_W18_C.tar.gz" MODEL_PATH "PaddleClas/v2.3/HRNet")
78+
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL_FOR_NNADAPTER_UNITTESTS} "PPLCNet_x0_25.tar.gz" MODEL_PATH "PaddleClas/v2.3/PPLCNet")
79+
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL_FOR_NNADAPTER_UNITTESTS} "Res2Net50_26w_4s.tar.gz" MODEL_PATH "PaddleClas/v2.3/Res2Net")
80+
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL_FOR_NNADAPTER_UNITTESTS} "SE_ResNet50_vd.tar.gz" MODEL_PATH "PaddleClas/v2.3/SENet")
81+
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL_FOR_NNADAPTER_UNITTESTS} "ViT_base_patch16_224.tar.gz" MODEL_PATH "PaddleClas/v2.3/VisionTransformer")
8282
# PaddleDetection
8383
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL_FOR_UNITTESTS} "ssd_mobilenet_v1_relu_voc_v2_2.tar.gz")
8484
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL_FOR_UNITTESTS} "ssd_vgg16_voc_v2_2.tar.gz")
@@ -90,13 +90,42 @@ if (WITH_TESTING)
9090
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL_FOR_UNITTESTS} "ppyolo_r50vd_dcn_1x_coco_v2_2.tar.gz")
9191
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL_FOR_UNITTESTS} "higherhrnet_hrnet_w32_640_v2_0.tar.gz")
9292
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL_FOR_UNITTESTS} "hrnet_w32_384x288_v2_0.tar.gz")
93+
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL_FOR_NNADAPTER_UNITTESTS} "picodet_m_416_coco.tar.gz" MODEL_PATH "PaddleDetection/v2.3")
94+
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL_FOR_NNADAPTER_UNITTESTS} "ppyolo_mbv3_large_coco.tar.gz" MODEL_PATH "PaddleDetection/v2.3")
95+
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL_FOR_NNADAPTER_UNITTESTS} "ppyolo_r50vd_dcn_1x_coco.tar.gz" MODEL_PATH "PaddleDetection/v2.3")
96+
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL_FOR_NNADAPTER_UNITTESTS} "ppyolo_tiny_650e_coco.tar.gz" MODEL_PATH "PaddleDetection/v2.3")
97+
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL_FOR_NNADAPTER_UNITTESTS} "ppyolov2_r50vd_dcn_365e_coco.tar.gz" MODEL_PATH "PaddleDetection/v2.3")
98+
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL_FOR_NNADAPTER_UNITTESTS} "tinypose_128x96.tar.gz" MODEL_PATH "PaddleDetection/v2.3")
99+
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL_FOR_NNADAPTER_UNITTESTS} "ssdlite_mobilenet_v3_small.tar.gz" MODEL_PATH "PaddleDetection/static")
100+
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL_FOR_NNADAPTER_UNITTESTS} "ssdlite_mobilenet_v3_large.tar.gz" MODEL_PATH "PaddleDetection/static")
101+
# PaddleSeg
102+
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL_FOR_NNADAPTER_UNITTESTS} "bisenet.tar.gz" MODEL_PATH "PaddleSeg/v2.3")
103+
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL_FOR_NNADAPTER_UNITTESTS} "segformer.tar.gz" MODEL_PATH "PaddleSeg/v2.3")
104+
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL_FOR_NNADAPTER_UNITTESTS} "stdcseg.tar.gz" MODEL_PATH "PaddleSeg/v2.3")
105+
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL_FOR_NNADAPTER_UNITTESTS} "unet_cityscapes_1024x512_160k.tar.gz" MODEL_PATH "PaddleSeg/v2.3")
106+
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL_FOR_NNADAPTER_UNITTESTS} "deeplabv3p_resnet50_os8_cityscapes_1024x512_80k.tar.gz" MODEL_PATH "PaddleSeg/v2.3")
107+
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL_FOR_NNADAPTER_UNITTESTS} "pphumanseg_lite_generic_192x192_with_softmax.tar.gz" MODEL_PATH "PaddleSeg/v2.3")
108+
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL_FOR_NNADAPTER_UNITTESTS} "deeplabv3p_resnet50_os8_humanseg_512x512_100k_with_softmax.tar.gz" MODEL_PATH "PaddleSeg/v2.3")
93109
# PaddleOCR
94-
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL_FOR_UNITTESTS} "ch_ppocr_mobile_v2_0_rec_v2_0.tar.gz")
95-
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL_FOR_UNITTESTS} "ch_ppocr_mobile_v2_0_det_v2_0.tar.gz")
110+
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL_FOR_NNADAPTER_UNITTESTS} "rec_crnn_mv3_ctc.tar.gz" MODEL_PATH "PaddleOCR")
111+
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL_FOR_NNADAPTER_UNITTESTS} "ch_ppocr_mobile_v2_0_rec_v2_0.tar.gz" MODEL_PATH "PaddleOCR/v2.3")
112+
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL_FOR_NNADAPTER_UNITTESTS} "ch_ppocr_mobile_v2_0_det_v2_0.tar.gz" MODEL_PATH "PaddleOCR/v2.3")
113+
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL_FOR_NNADAPTER_UNITTESTS} "ch_ppocr_server_v2.0_rec_infer.tar.gz" MODEL_PATH "PaddleOCR/v2.3")
114+
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL_FOR_NNADAPTER_UNITTESTS} "ch_ppocr_server_v2.0_det_infer.tar.gz" MODEL_PATH "PaddleOCR/v2.3")
115+
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL_FOR_NNADAPTER_UNITTESTS} "ch_PP-OCRv2_det_infer.tar.gz" MODEL_PATH "PaddleOCR/v2.3")
116+
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL_FOR_NNADAPTER_UNITTESTS} "ch_PP-OCRv2_rec_infer.tar.gz" MODEL_PATH "PaddleOCR/v2.3")
117+
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL_FOR_NNADAPTER_UNITTESTS} "e2e_server_pgnetA.tar.gz" MODEL_PATH "PaddleOCR/v2.3")
96118
# PaddleNLP
97-
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL_FOR_UNITTESTS} "ernie_tiny_v2_0.tar.gz")
119+
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL_FOR_NNADAPTER_UNITTESTS} "ernie_tiny.tar.gz" MODEL_PATH "PaddleNLP")
98120
# PaddleGAN
99-
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL_FOR_UNITTESTS} "esrgan_psnr_x4_div2k_v2_0.tar.gz")
121+
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL_FOR_NNADAPTER_UNITTESTS} "esrgan_psnr_x4_div2k.tar.gz" MODEL_PATH "PaddleGAN")
122+
# PaddleRec
123+
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL_FOR_NNADAPTER_UNITTESTS} "deepfm.tar.gz" MODEL_PATH "PaddleRec/v2.1.0")
124+
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL_FOR_NNADAPTER_UNITTESTS} "naml.tar.gz" MODEL_PATH "PaddleRec/v2.1.0")
125+
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL_FOR_NNADAPTER_UNITTESTS} "ncf.tar.gz" MODEL_PATH "PaddleRec/v2.1.0")
126+
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL_FOR_NNADAPTER_UNITTESTS} "wide_deep.tar.gz" MODEL_PATH "PaddleRec/v2.1.0")
127+
# PaddleVideo
128+
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL_FOR_NNADAPTER_UNITTESTS} "ppTSN.tar.gz" MODEL_PATH "PaddleVideo/v2.2.0")
100129
endif()
101130
if(LITE_WITH_ARM)
102131
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL} "mobilenet_v1_int16.tar.gz")
@@ -143,9 +172,11 @@ if (WITH_TESTING)
143172
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL_FOR_UNITTESTS} "COCO2017_100.tar.gz")
144173
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL_FOR_UNITTESTS} "ocr_db_data.tar.gz")
145174
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL_FOR_UNITTESTS} "ocr_rec_data.tar.gz")
175+
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL_FOR_UNITTESTS} "ocr_e2e_data.tar.gz")
146176
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL_FOR_UNITTESTS} "ernie_tiny_data.tar.gz")
147177
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL_FOR_UNITTESTS} "COCO2017_keypoints_100.tar.gz")
148178
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL_FOR_UNITTESTS} "GAN_set14.tar.gz")
179+
lite_download_and_uncompress(${LITE_MODEL_DIR} ${LITE_URL_FOR_UNITTESTS} "specific_model_input_output.tar.gz")
149180
endif()
150181
endif()
151182

‎lite/tests/api/CMakeLists.txt

Lines changed: 33 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -142,13 +142,42 @@ if(LITE_WITH_NNADAPTER)
142142
lite_cc_test_with_model_and_data(test_yolov3_r50vd_dcn_coco_fp32_v2_2_nnadapter SOURCE test_yolov3_r50vd_dcn_coco_fp32_v2_2_nnadapter.cc MODEL yolov3_r50vd_dcn_coco_v2_2 DATA COCO2017_100)
143143
lite_cc_test_with_model_and_data(test_yolov4_cspdarknet_coco_fp32_v1_8_nnadapter SOURCE test_yolov4_cspdarknet_coco_fp32_v1_8_nnadapter.cc MODEL yolov4_cspdarknet_coco_v1_8 DATA COCO2017_100)
144144
lite_cc_test_with_model_and_data(test_ppyolo_r50vd_dcn_1x_coco_fp32_v2_2_nnadapter SOURCE test_ppyolo_r50vd_dcn_1x_coco_fp32_v2_2_nnadapter.cc MODEL ppyolo_r50vd_dcn_1x_coco_v2_2 DATA COCO2017_100)
145+
lite_cc_test_with_model_and_data(test_ppyolo_r50vd_dcn_1x_coco_fp32_v2_3_nnadapter SOURCE test_ppyolo_r50vd_dcn_1x_coco_fp32_v2_3_nnadapter.cc MODEL PaddleDetection/v2.3/ppyolo_r50vd_dcn_1x_coco DATA COCO2017_100)
146+
lite_cc_test_with_model_and_data(test_ppyolo_mbv3_large_coco_fp32_v2_3_nnadapter SOURCE test_ppyolo_mbv3_large_coco_fp32_v2_3_nnadapter.cc MODEL PaddleDetection/v2.3/ppyolo_mbv3_large_coco DATA COCO2017_100)
147+
lite_cc_test_with_model_and_data(test_ppyolo_tiny_coco_fp32_v2_3_nnadapter SOURCE test_ppyolo_tiny_coco_fp32_v2_3_nnadapter.cc MODEL PaddleDetection/v2.3/ppyolo_tiny_650e_coco DATA COCO2017_100)
148+
lite_cc_test_with_model_and_data(test_ppyolo2_r50vd_dcn_coco_fp32_v2_3_nnadapter SOURCE test_ppyolo2_r50vd_dcn_coco_fp32_v2_3_nnadapter.cc MODEL PaddleDetection/v2.3/ppyolov2_r50vd_dcn_365e_coco DATA COCO2017_100)
149+
lite_cc_test_with_model_and_data(test_picodet_coco_fp32_v2_3_nnadapter SOURCE test_picodet_coco_fp32_v2_3_nnadapter.cc MODEL PaddleDetection/v2.3/picodet_m_416_coco DATA COCO2017_100)
150+
lite_cc_test_with_model_and_data(test_ssdlite_mobilenet_v3_large_coco_fp32_static_nnadapter SOURCE test_ssdlite_mobilenet_v3_large_coco_fp32_static_nnadapter.cc MODEL PaddleDetection/static/ssdlite_mobilenet_v3_small DATA COCO2017_100)
151+
lite_cc_test_with_model_and_data(test_ssdlite_mobilenet_v3_small_coco_fp32_static_nnadapter SOURCE test_ssdlite_mobilenet_v3_small_coco_fp32_static_nnadapter.cc MODEL PaddleDetection/static/ssdlite_mobilenet_v3_large DATA COCO2017_100)
145152
lite_cc_test_with_model_and_data(test_higher_hrnet_w32_640_fp32_v2_0_nnadapter SOURCE test_higher_hrnet_w32_640_fp32_v2_0_nnadapter.cc MODEL higherhrnet_hrnet_w32_640_v2_0 DATA COCO2017_keypoints_100)
146153
lite_cc_test_with_model_and_data(test_hrnet_w32_384x288_fp32_v2_0_nnadapter SOURCE test_hrnet_w32_384x288_fp32_v2_0_nnadapter.cc MODEL hrnet_w32_384x288_v2_0 DATA COCO2017_keypoints_100)
154+
lite_cc_test_with_model_and_data(test_pptinypose_fp32_v2_3_nnadapter SOURCE test_pptinypose_fp32_v2_3_nnadapter.cc MODEL PaddleDetection/v2.3/tinypose_128x96 DATA specific_model_input_output)
147155
# PaddleOCR
148-
lite_cc_test_with_model_and_data(test_ch_ppocr_mobile_v2_0_det_fp32_v2_0_nnadapter SOURCE test_ch_ppocr_mobile_v2_0_det_fp32_v2_0_nnadapter.cc MODEL ch_ppocr_mobile_v2_0_det_v2_0 DATA ocr_db_data)
149-
lite_cc_test_with_model_and_data(test_ch_ppocr_mobile_v2_0_rec_fp32_v2_0_nnadapter SOURCE test_ch_ppocr_mobile_v2_0_rec_fp32_v2_0_nnadapter.cc MODEL ch_ppocr_mobile_v2_0_rec_v2_0 DATA ocr_rec_data)
156+
lite_cc_test_with_model_and_data(test_ch_ppocr_mobile_v2_0_det_fp32_v2_3_nnadapter SOURCE test_ch_ppocr_mobile_v2_0_det_fp32_v2_3_nnadapter.cc MODEL PaddleOCR/v2.3/ch_ppocr_mobile_v2_0_det_v2_0 DATA ocr_db_data)
157+
lite_cc_test_with_model_and_data(test_ch_ppocr_mobile_v2_0_rec_fp32_v2_3_nnadapter SOURCE test_ch_ppocr_mobile_v2_0_rec_fp32_v2_3_nnadapter.cc MODEL PaddleOCR/v2.3/ch_ppocr_mobile_v2_0_rec_v2_0 DATA ocr_rec_data)
158+
lite_cc_test_with_model_and_data(test_ch_ppocr_server_v2_0_det_fp32_v2_3_nnadapter SOURCE test_ch_ppocr_server_v2_0_det_fp32_v2_3_nnadapter.cc MODEL PaddleOCR/v2.3/ch_ppocr_server_v2.0_det_infer DATA ocr_db_data)
159+
lite_cc_test_with_model_and_data(test_ch_ppocr_server_v2_0_rec_fp32_v2_3_nnadapter SOURCE test_ch_ppocr_server_v2_0_rec_fp32_v2_3_nnadapter.cc MODEL PaddleOCR/v2.3/ch_ppocr_server_v2.0_rec_infer DATA ocr_rec_data)
160+
lite_cc_test_with_model_and_data(test_ch_ppocr_v2_det_fp32_v2_3_nnadapter SOURCE test_ch_ppocr_v2_det_fp32_v2_3_nnadapter.cc MODEL PaddleOCR/v2.3/ch_PP-OCRv2_det_infer DATA ocr_db_data)
161+
lite_cc_test_with_model_and_data(test_ch_ppocr_v2_rec_fp32_v2_3_nnadapter SOURCE test_ch_ppocr_v2_rec_fp32_v2_3_nnadapter.cc MODEL PaddleOCR/v2.3/ch_PP-OCRv2_rec_infer DATA ocr_rec_data)
162+
lite_cc_test_with_model_and_data(test_crnn_ctc_fp32_v2_3_nnadapter SOURCE test_crnn_ctc_fp32_v2_3_nnadapter.cc MODEL PaddleOCR/rec_crnn_mv3_ctc DATA ocr_rec_data)
163+
lite_cc_test_with_model_and_data(test_e2e_server_pgnetA_fp32_v2_3_nnadapter SOURCE test_e2e_server_pgnetA_fp32_v2_3_nnadapter.cc MODEL PaddleOCR/v2.3/e2e_server_pgnetA DATA ocr_e2e_data)
150164
# PaddleNLP
151-
lite_cc_test_with_model_and_data(test_ernie_tiny_fp32_v2_0_nnadapter SOURCE test_ernie_tiny_fp32_v2_0_nnadapter.cc MODEL ernie_tiny_v2_0 DATA ernie_tiny_data)
165+
lite_cc_test_with_model_and_data(test_ernie_tiny_fp32_v2_0_nnadapter SOURCE test_ernie_tiny_fp32_v2_0_nnadapter.cc MODEL PaddleNLP/ernie_tiny DATA ernie_tiny_data)
152166
# PaddleGAN
153-
lite_cc_test_with_model_and_data(test_esrgan_psnr_x4_div2k_fp32_v2_0_nnadapter SOURCE test_esrgan_psnr_x4_div2k_fp32_v2_0_nnadapter.cc MODEL esrgan_psnr_x4_div2k_v2_0 DATA GAN_set14)
167+
lite_cc_test_with_model_and_data(test_esrgan_psnr_x4_div2k_fp32_v2_0_nnadapter SOURCE test_esrgan_psnr_x4_div2k_fp32_v2_0_nnadapter.cc MODEL PaddleGAN/esrgan_psnr_x4_div2k DATA GAN_set14)
168+
# PaddleRec
169+
lite_cc_test_with_model_and_data(test_naml_fp32_v2_1_nnadapter SOURCE test_naml_fp32_v2_1_nnadapter.cc MODEL PaddleRec/v2.1.0/naml DATA specific_model_input_output)
170+
lite_cc_test_with_model_and_data(test_deepfm_fp32_v2_1_nnadapter SOURCE test_deepfm_fp32_v2_1_nnadapter.cc MODEL PaddleRec/v2.1.0/deepfm DATA specific_model_input_output)
171+
lite_cc_test_with_model_and_data(test_ncf_fp32_v2_1_nnadapter SOURCE test_ncf_fp32_v2_1_nnadapter.cc MODEL PaddleRec/v2.1.0/ncf DATA specific_model_input_output)
172+
lite_cc_test_with_model_and_data(test_wide_deep_fp32_v2_1_nnadapter SOURCE test_wide_deep_fp32_v2_1_nnadapter.cc MODEL PaddleRec/v2.1.0/wide_deep DATA specific_model_input_output)
173+
# PaddleSeg
174+
lite_cc_test_with_model_and_data(test_segformer_fp32_v2_3_nnadapter SOURCE test_segformer_fp32_v2_3_nnadapter.cc MODEL PaddleSeg/v2.3/segformer DATA specific_model_input_output)
175+
lite_cc_test_with_model_and_data(test_bisenet_fp32_v2_3_nnadapter SOURCE test_bisenet_fp32_v2_3_nnadapter.cc MODEL PaddleSeg/v2.3/bisenet DATA specific_model_input_output)
176+
lite_cc_test_with_model_and_data(test_stdcseg_fp32_v2_3_nnadapter SOURCE test_stdcseg_fp32_v2_3_nnadapter.cc MODEL PaddleSeg/v2.3/stdcseg DATA specific_model_input_output)
177+
lite_cc_test_with_model_and_data(test_pphumanseg_lite_fp32_v2_3_nnadapter SOURCE test_pphumanseg_lite_fp32_v2_3_nnadapter.cc MODEL PaddleSeg/v2.3/pphumanseg_lite_generic_192x192_with_softmax DATA specific_model_input_output)
178+
lite_cc_test_with_model_and_data(test_unet_cityscapes_fp32_v2_3_nnadapter SOURCE test_unet_cityscapes_fp32_v2_3_nnadapter.cc MODEL PaddleSeg/v2.3/unet_cityscapes_1024x512_160k DATA specific_model_input_output)
179+
lite_cc_test_with_model_and_data(test_deeplabv3p_resnet50_os8_cityscapes_fp32_v2_3_nnadapter SOURCE test_deeplabv3p_resnet50_os8_cityscapes_fp32_v2_3_nnadapter.cc MODEL PaddleSeg/v2.3/deeplabv3p_resnet50_os8_cityscapes_1024x512_80k DATA specific_model_input_output)
180+
lite_cc_test_with_model_and_data(test_deeplabv3p_resnet50_os8_humanseg_fp32_v2_3_nnadapter SOURCE test_deeplabv3p_resnet50_os8_humanseg_fp32_v2_3_nnadapter.cc MODEL PaddleSeg/v2.3/deeplabv3p_resnet50_os8_humanseg_512x512_100k_with_softmax DATA specific_model_input_output)
181+
# PaddleVideo
182+
lite_cc_test_with_model_and_data(test_pptsn_fp32_v2_2_nnadapter SOURCE test_pptsn_fp32_v2_2_nnadapter.cc MODEL PaddleVideo/v2.2.0/ppTSN DATA specific_model_input_output)
154183
endif()

‎lite/tests/api/COCO2017_utility.h

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -48,17 +48,13 @@ std::vector<std::vector<T>> ReadRawData(
4848
std::ifstream fin(raw_data_file_dir, std::ios::in | std::ios::binary);
4949
CHECK(fin.is_open()) << "failed to open file " << raw_data_file_dir;
5050
fin.seekg(0, std::ios::end);
51-
int file_size = fin.tellg();
5251
fin.seekg(0, std::ios::beg);
53-
CHECK_EQ(static_cast<size_t>(file_size),
54-
static_cast<size_t>(image_size) * sizeof(T) / sizeof(char));
55-
fin.read(reinterpret_cast<char*>(data), file_size);
52+
fin.read(reinterpret_cast<char*>(data), image_size * sizeof(T));
5653
fin.close();
5754
data += image_size;
5855
}
5956
raw_data.emplace_back(one_iter_raw_data);
6057
}
61-
6258
return raw_data;
6359
}
6460

0 commit comments

Comments
 (0)