test/AIAS
1
0
Fork 0
mirror of https://gitee.com/mymagicpower/AIAS.git synced 2024-11-29 18:58:16 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

no message

Browse Source
This commit is contained in:
Calvin 2022-06-30 11:49:51 +08:00
parent 838200ab8f
commit b910488451
49 changed files with 49 additions and 49 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
1_image_sdks/embedding/feature_extraction_sdk/src/main/java/me/aias/util/ImageEncoderModel.java
View File

@ -28,7 +28,7 @@ public final class ImageEncoderModel {
//.optModelUrls("/Users/calvin/CLIP-ViT-B-32-IMAGE/CLIP-ViT-B-32-IMAGE.zip")
.optTranslator(new ImageTranslator())
.optEngine("PyTorch") // Use PyTorch engine
.optOption("mapLocation", "true")
.optDevice(Device.cpu())
.optProgress(new ProgressBar())
.build();

2
1_image_sdks/gan/style_transfer_sdk/src/main/java/me/aias/util/StyleTransfer.java
View File

@ -28,7 +28,7 @@ public final class StyleTransfer {
.optEngine("PyTorch") // Use PyTorch engine
.optModelUrls(modelUrl)
.optProgress(new ProgressBar())
.optOption("mapLocation", "true")
.optDevice(Device.cpu())
.optTranslatorFactory(new StyleTransferTranslatorFactory())
.build();

2
1_image_sdks/security/face_sdks/face_align_sdk/src/main/java/me/aias/util/LightFaceDetection.java
View File

@ -27,7 +27,7 @@ public final class LightFaceDetection {
.optTranslator(translator)
.optProgress(new ProgressBar())
.optEngine("PyTorch") // Use PyTorch engine
.optOption("mapLocation", "true")
.optDevice(Device.cpu())
.build();
return criteria;

2
1_image_sdks/security/face_sdks/face_align_sdk/src/main/java/me/aias/util/RetinaFaceDetection.java
View File

@ -27,7 +27,7 @@ public final class RetinaFaceDetection {
.optTranslator(translator)
.optProgress(new ProgressBar())
.optEngine("PyTorch") // Use PyTorch engine
.optOption("mapLocation", "true")
.optDevice(Device.cpu())
.build();
return criteria;

2
1_image_sdks/security/face_sdks/face_feature_sdk/src/main/java/me/aias/util/FaceFeature.java
View File

@ -19,7 +19,7 @@ public final class FaceFeature {
.optTranslator(new FaceFeatureTranslator())
.optProgress(new ProgressBar())
.optEngine("PyTorch") // Use PyTorch engine
.optOption("mapLocation", "true")
.optDevice(Device.cpu())
.build();
return criteria;

2
1_image_sdks/security/face_sdks/retinaface_sdk/src/main/java/me/aias/example/utils/LightFaceDetection.java
View File

@ -27,7 +27,7 @@ public final class LightFaceDetection {
.optTranslator(translator)
.optProgress(new ProgressBar())
.optEngine("PyTorch") // Use PyTorch engine
.optOption("mapLocation", "true")
.optDevice(Device.cpu())
.build();
return criteria;

2
1_image_sdks/security/face_sdks/retinaface_sdk/src/main/java/me/aias/example/utils/RetinaFaceDetection.java
View File

@ -27,7 +27,7 @@ public final class RetinaFaceDetection {
.optTranslator(translator)
.optProgress(new ProgressBar())
.optEngine("PyTorch") // Use PyTorch engine
.optOption("mapLocation", "true")
.optDevice(Device.cpu())
.build();
return criteria;

2
1_image_sdks/security/fire_smoke_sdk/src/main/java/me/aias/example/utils/FireSmokeDetect.java
View File

@ -41,7 +41,7 @@ public final class FireSmokeDetect {
.optTranslator(translator)
.optProgress(new ProgressBar())
.optEngine("PyTorch")
.optOption("mapLocation", "true")
.optDevice(Device.cpu())
.build();
return criteria;

2
1_image_sdks/security/reflective_vest_sdk/src/main/java/me/aias/example/utils/LargeHelmetHeadPersonDetect.java
View File

@ -42,7 +42,7 @@ public final class LargeHelmetHeadPersonDetect {
.optTranslator(translator)
.optProgress(new ProgressBar())
.optEngine("PyTorch")
.optOption("mapLocation", "true")
.optDevice(Device.cpu())
.build();
return criteria;

2
1_image_sdks/security/reflective_vest_sdk/src/main/java/me/aias/example/utils/ReflectiveVest.java
View File

@ -42,7 +42,7 @@ public final class ReflectiveVest {
.optTranslator(translator)
.optProgress(new ProgressBar())
.optEngine("PyTorch")
.optOption("mapLocation", "true")
.optDevice(Device.cpu())
.build();
return criteria;

2
1_image_sdks/security/smart_construction_sdk/src/main/java/me/aias/example/utils/Yolov5lDetect.java
View File

@ -42,7 +42,7 @@ public final class Yolov5lDetect {
.optTranslator(translator)
.optProgress(new ProgressBar())
.optEngine("PyTorch")
.optOption("mapLocation", "true")
.optDevice(Device.cpu())
.build();
return criteria;

2
1_image_sdks/security/smart_construction_sdk/src/main/java/me/aias/example/utils/Yolov5mDetect.java
View File

@ -42,7 +42,7 @@ public final class Yolov5mDetect {
.optTranslator(translator)
.optProgress(new ProgressBar())
.optEngine("PyTorch")
.optOption("mapLocation", "true")
.optDevice(Device.cpu())
.build();
return criteria;

2
1_image_sdks/security/smart_construction_sdk/src/main/java/me/aias/example/utils/Yolov5sDetect.java
View File

@ -42,7 +42,7 @@ public final class Yolov5sDetect {
.optTranslator(translator)
.optProgress(new ProgressBar())
.optEngine("PyTorch")
.optOption("mapLocation", "true")
.optDevice(Device.cpu())
.build();
return criteria;

2
2_nlp_sdks/embedding/sentence_encoder_100_sdk/src/main/java/me/aias/example/utils/SentenceEncoder.java
View File

@ -22,7 +22,7 @@ public final class SentenceEncoder {
// .optModelUrls("/Users/calvin/models/paraphrase-xlm-r-multilingual-v1/")
.optTranslator(new SentenceTransTranslator(processor))
.optEngine("PyTorch") // Use PyTorch engine
.optOption("mapLocation", "true")
.optDevice(Device.cpu())
.optProgress(new ProgressBar())
.build();

2
2_nlp_sdks/embedding/sentence_encoder_15_sdk/src/main/java/me/aias/example/utils/SentenceEncoder.java
View File

@ -22,7 +22,7 @@ public final class SentenceEncoder {
// .optModelUrls("/Users/calvin/models/distiluse-base-multilingual-cased-v1/")
.optTranslator(new SentenceTransTranslator())
.optEngine("PyTorch") // Use PyTorch engine
.optOption("mapLocation", "true")
.optDevice(Device.cpu())
.optProgress(new ProgressBar())
.build();

2
2_nlp_sdks/embedding/sentence_encoder_en_sdk/src/main/java/me/aias/example/utils/SentenceEncoderEN.java
View File

@ -22,7 +22,7 @@ public final class SentenceEncoderEN {
// .optModelUrls("/Users/calvin/msmarco-MiniLM-L-6-v3/")
.optTranslator(new SentenceTransTranslator())
.optEngine("PyTorch") // Use PyTorch engine
.optOption("mapLocation", "true")
.optDevice(Device.cpu())
.optProgress(new ProgressBar())
.build();

2
2_nlp_sdks/image_text_search/image_text_search_40_sdk/src/main/java/me/aias/example/utils/ImageEncoder.java
View File

@ -23,7 +23,7 @@ public final class ImageEncoder {
// .optModelUrls("/Users/calvin/CLIP-ViT-B-32-IMAGE/")
.optTranslator(new ImageTranslator())
.optEngine("PyTorch") // Use PyTorch engine
.optOption("mapLocation", "true")
.optDevice(Device.cpu())
.optProgress(new ProgressBar())
.build();

2
2_nlp_sdks/image_text_search/image_text_search_40_sdk/src/main/java/me/aias/example/utils/TextEncoder.java
View File

@ -18,7 +18,7 @@ public final class TextEncoder {
// .optModelUrls("/Users/calvin/M-BERT-Base-ViT-B/")
.optTranslator(new TextTranslator(isChinese))
.optEngine("PyTorch") // Use PyTorch engine
.optOption("mapLocation", "true")
.optDevice(Device.cpu())
.optProgress(new ProgressBar())
.build();

2
2_nlp_sdks/image_text_search/image_text_search_cn_sdk/src/main/java/me/aias/example/utils/ImageEncoder.java
View File

@ -24,7 +24,7 @@ public final class ImageEncoder {
// .optModelUrls("/Users/calvin/CLIP-ViT-B-32-IMAGE/")
.optTranslator(new ImageTranslator())
.optEngine("PyTorch") // Use PyTorch engine
.optOption("mapLocation", "true")
.optDevice(Device.cpu())
.optProgress(new ProgressBar())
.build();

2
2_nlp_sdks/image_text_search/image_text_search_cn_sdk/src/main/java/me/aias/example/utils/TextEncoder.java
View File

@ -23,7 +23,7 @@ public final class TextEncoder {
// .optModelUrls("/Users/calvin/CLIP-ViT-B-32-TEXT/")
.optTranslator(new TextTranslator())
.optEngine("PyTorch") // Use PyTorch engine
.optOption("mapLocation", "true")
.optDevice(Device.cpu())
.optProgress(new ProgressBar())
.build();

2
2_nlp_sdks/image_text_search/image_text_search_sdk/src/main/java/me/aias/example/utils/ImageEncoder.java
View File

@ -23,7 +23,7 @@ public final class ImageEncoder {
// .optModelUrls("/Users/calvin/CLIP-ViT-B-32-IMAGE/")
.optTranslator(new ImageTranslator())
.optEngine("PyTorch") // Use PyTorch engine
.optOption("mapLocation", "true")
.optDevice(Device.cpu())
.optProgress(new ProgressBar())
.build();

2
2_nlp_sdks/image_text_search/image_text_search_sdk/src/main/java/me/aias/example/utils/TextEncoder.java
View File

@ -22,7 +22,7 @@ public final class TextEncoder {
// .optModelUrls("/Users/calvin/CLIP-ViT-B-32-TEXT/")
.optTranslator(new TextTranslator())
.optEngine("PyTorch") // Use PyTorch engine
.optOption("mapLocation", "true")
.optDevice(Device.cpu())
.optProgress(new ProgressBar())
.build();

2
2_nlp_sdks/semantic_search/cross_encoder_en_sdk/src/main/java/me/aias/example/utils/TinyBertCrossEncoder.java
View File

@ -20,7 +20,7 @@ public final class TinyBertCrossEncoder {
// .optModelUrls("/Users/calvin/ms-marco-TinyBERT-L-2-v2/")
.optTranslator(new CrossEncoderTranslator())
.optEngine("PyTorch") // Use PyTorch engine
.optOption("mapLocation", "true")
.optDevice(Device.cpu())
.optProgress(new ProgressBar())
.build();

2
2_nlp_sdks/semantic_search/qa_natural_questions_sdk/src/main/java/me/aias/example/utils/QANaturalQuestions.java
View File

@ -22,7 +22,7 @@ public final class QANaturalQuestions {
// .optModelUrls("/Users/calvin/nq-distilbert-base-v1/")
.optTranslator(new QATranslator())
.optEngine("PyTorch") // Use PyTorch engine
.optOption("mapLocation", "true")
.optDevice(Device.cpu())
.optProgress(new ProgressBar())
.build();

2
2_nlp_sdks/semantic_search/qa_retrieval_msmarco_s_sdk/src/main/java/me/aias/example/utils/QuestionAnswerRetrieval.java
View File

@ -22,7 +22,7 @@ public final class QuestionAnswerRetrieval {
// .optModelUrls("/Users/calvin/msmarco-MiniLM-L-6-v3/")
.optTranslator(new SentenceTransTranslator())
.optEngine("PyTorch") // Use PyTorch engine
.optOption("mapLocation", "true")
.optDevice(Device.cpu())
.optProgress(new ProgressBar())
.build();

2
2_nlp_sdks/semantic_search/semantic_search_publications_sdk/src/main/java/me/aias/example/utils/SemanticSearchPublications.java
View File

@ -22,7 +22,7 @@ public final class SemanticSearchPublications {
// .optModelUrls("/Users/calvin/allenai-specter/")
.optTranslator(new SemanticSearchTranslator())
.optEngine("PyTorch") // Use PyTorch engine
.optOption("mapLocation", "true")
.optDevice(Device.cpu())
.optProgress(new ProgressBar())
.build();

2
2_nlp_sdks/sentiment_analysis/sentiment_analysis_sdk/src/main/java/me/aias/example/utils/SentimentAnalysis.java
View File

@ -24,7 +24,7 @@ public final class SentimentAnalysis {
// "/Users/calvin/Documents/build/pytorch_models/distilbert_sst_english/")
.optTranslator(new PtDistilBertTranslator())
.optEngine("PyTorch") // Use PyTorch engine
.optOption("mapLocation", "true")
.optDevice(Device.cpu())
.optProgress(new ProgressBar())
.build();

2
3_audio_sdks/sv2tts_speakencoder_sdk/src/main/java/me/aias/example/utils/SpeakerEncoder.java
View File

@ -18,7 +18,7 @@ public class SpeakerEncoder {
// .optModelPath(Paths.get("src/main/resources/voice/tacotronSTFT.pt"))
.optTranslator(new SpeakerEncoderTranslator())
.optEngine("PyTorch") // Use PyTorch engine
.optOption("mapLocation", "true")
.optDevice(Device.cpu())
.optProgress(new ProgressBar())
.build();

2
3_audio_sdks/sv2tts_waveglow_sdk/src/main/java/me/aias/example/speaker/SpeakerEncoder.java
View File

@ -18,7 +18,7 @@ public class SpeakerEncoder {
// .optModelPath(Paths.get("src/main/resources/voice/tacotronSTFT.pt"))
.optTranslator(new SpeakerEncoderTranslator())
.optEngine("PyTorch") // Use PyTorch engine
.optOption("mapLocation", "true")
.optDevice(Device.cpu())
.optProgress(new ProgressBar())
.build();

2
3_audio_sdks/sv2tts_waveglow_sdk/src/main/java/me/aias/example/tacotron2/Tacotron2Encoder.java
View File

@ -16,7 +16,7 @@ public class Tacotron2Encoder {
.optModelUrls("https://aias-home.oss-cn-beijing.aliyuncs.com/models/speech_models/tacotron2.zip")
.optTranslator(new TacotronTranslator())
.optEngine("PyTorch") // Use PyTorch engine
.optOption("mapLocation", "true")
.optDevice(Device.cpu())
.optProgress(new ProgressBar())
.build();

2
3_audio_sdks/sv2tts_waveglow_sdk/src/main/java/me/aias/example/utils/DenoiserEncoder.java
View File

@ -17,7 +17,7 @@ public class DenoiserEncoder {
"https://aias-home.oss-cn-beijing.aliyuncs.com/models/speech_models/denoiser.zip")
.optTranslator(new DenoiserTranslator())
.optEngine("PyTorch") // Use PyTorch engine
.optOption("mapLocation", "true")
.optDevice(Device.cpu())
.optProgress(new ProgressBar())
.build();

2
3_audio_sdks/sv2tts_waveglow_sdk/src/main/java/me/aias/example/utils/WaveGlowEncoder.java
View File

@ -17,7 +17,7 @@ public class WaveGlowEncoder {
"https://aias-home.oss-cn-beijing.aliyuncs.com/models/speech_models/waveGlow.zip")
.optTranslator(new WaveGlowTranslator())
.optEngine("PyTorch") // Use PyTorch engine
.optOption("mapLocation", "true")
.optDevice(Device.cpu())
.optProgress(new ProgressBar())
.build();

2
3_audio_sdks/tacotron2_sdk/src/main/java/me/aias/example/speaker/SpeakerEncoder.java
View File

@ -18,7 +18,7 @@ public class SpeakerEncoder {
// .optModelPath(Paths.get("src/main/resources/voice/tacotronSTFT.pt"))
.optTranslator(new SpeakerEncoderTranslator())
.optEngine("PyTorch") // Use PyTorch engine
.optOption("mapLocation", "true")
.optDevice(Device.cpu())
.optProgress(new ProgressBar())
.build();

2
3_audio_sdks/tacotron2_sdk/src/main/java/me/aias/example/tacotron2/Tacotron2Encoder.java
View File

@ -16,7 +16,7 @@ public class Tacotron2Encoder {
.optModelUrls("https://aias-home.oss-cn-beijing.aliyuncs.com/models/speech_models/tacotron2.zip")
.optTranslator(new TacotronTranslator())
.optEngine("PyTorch") // Use PyTorch engine
.optOption("mapLocation", "true")
.optDevice(Device.cpu())
.optProgress(new ProgressBar())
.build();

2
3_audio_sdks/tacotron_stft_sdk/src/main/java/me/aias/example/utils/Tacotron.java
View File

@ -19,7 +19,7 @@ public class Tacotron {
// .optModelPath(Paths.get("src/main/resources/voice/tacotronSTFT.pt"))
.optTranslator(new TacotronTranslator())
.optEngine("PyTorch") // Use PyTorch engine
.optOption("mapLocation", "true")
.optDevice(Device.cpu())
.optProgress(new ProgressBar())
.build();

2
5_bigdata_sdks/flink_sentence_encoder_sdk/src/main/java/me/aias/example/utils/SentenceEncoder.java
View File

@ -22,7 +22,7 @@ public final class SentenceEncoder {
// .optModelUrls("/Users/calvin/models/distiluse-base-multilingual-cased-v1/")
.optTranslator(new SentenceTransTranslator())
.optEngine("PyTorch") // Use PyTorch engine
.optOption("mapLocation", "true")
.optDevice(Device.cpu())
.optProgress(new ProgressBar())
.build();

2
5_bigdata_sdks/flink_sentiment_analysis_sdk/src/main/java/me/aias/example/utils/SentimentAnalysis.java
View File

@ -24,7 +24,7 @@ public final class SentimentAnalysis {
// "/Users/calvin/Documents/build/pytorch_models/distilbert_sst_english/")
.optTranslator(new PtDistilBertTranslator())
.optEngine("PyTorch") // Use PyTorch engine
.optOption("mapLocation", "true")
.optDevice(Device.cpu())
.optProgress(new ProgressBar())
.build();

2
5_bigdata_sdks/kafka_sentence_encoder_sdk/src/main/java/me/aias/example/utils/SentenceEncoder.java
View File

@ -22,7 +22,7 @@ public final class SentenceEncoder {
// .optModelUrls("/Users/calvin/models/distiluse-base-multilingual-cased-v1/")
.optTranslator(new SentenceTransTranslator())
.optEngine("PyTorch") // Use PyTorch engine
.optOption("mapLocation", "true")
.optDevice(Device.cpu())
.optProgress(new ProgressBar())
.build();

2
5_bigdata_sdks/kafka_sentiment_analysis_sdk/src/main/java/me/aias/example/SentimentAnalysis.java
View File

@ -24,7 +24,7 @@ public final class SentimentAnalysis {
// "/Users/calvin/Documents/build/pytorch_models/distilbert_sst_english/")
.optTranslator(new PtDistilBertTranslator())
.optEngine("PyTorch") // Use PyTorch engine
.optOption("mapLocation", "true")
.optDevice(Device.cpu())
.optProgress(new ProgressBar())
.build();

2
7_engine_hub/api_platform/api-platform/src/main/java/me/aias/infer/face/FaceDetectionModel.java
View File

@ -52,7 +52,7 @@ public final class FaceDetectionModel {
.optModelUrls(layoutUri)
.optTranslator(translator)
.optEngine("PyTorch") // Use PyTorch engine
.optOption("mapLocation", "true")
.optDevice(Device.cpu())
.optProgress(new ProgressBar())
.build();
return criteria;

2
7_engine_hub/api_platform/api-platform/src/main/java/me/aias/infer/face/FaceFeatureModel.java
View File

@ -45,7 +45,7 @@ public final class FaceFeatureModel {
.optModelUrls(layoutUri)
.optTranslator(new FaceFeatureTranslator())
.optEngine("PyTorch") // Use PyTorch engine
.optOption("mapLocation", "true")
.optDevice(Device.cpu())
.optProgress(new ProgressBar())
.build();
return criteria;

2
7_engine_hub/image_text_search/image-text-search/src/main/java/me/aias/model/ImageEncoderModel.java
View File

@ -41,7 +41,7 @@ public final class ImageEncoderModel {
.optTranslator(new ImageTranslator())
.optEngine("PyTorch") // Use PyTorch engine
.optDevice(Device.cpu())
// .optOption("mapLocation", "true")
// .optDevice(Device.cpu())
.optProgress(new ProgressBar())
.build();

2
7_engine_hub/image_text_search/image-text-search/src/main/java/me/aias/model/TextEncoderModel.java
View File

@ -40,7 +40,7 @@ public final class TextEncoderModel {
.optTranslator(new TextTranslator(isChinese))
.optEngine("PyTorch") // Use PyTorch engine
.optDevice(Device.cpu())
// .optOption("mapLocation", "true")
// .optDevice(Device.cpu())
.optProgress(new ProgressBar())
.build();

2
7_engine_hub/text_search/text-search/src/main/java/me/aias/common/sentence/SentenceEncoderModel.java
View File

@ -44,7 +44,7 @@ public final class SentenceEncoderModel {
.optTranslator(new SentenceTransTranslator())
.optEngine("PyTorch") // Use PyTorch engine
.optDevice(Device.cpu())
// .optOption("mapLocation", "true")
// .optDevice(Device.cpu())
.optProgress(new ProgressBar())
.build();
return criteria;

2
7_engine_hub/video_search/video-search/src/main/java/me/aias/common/face/FaceFeatureModel.java
View File

@ -39,7 +39,7 @@ public final class FaceFeatureModel {
.optTranslator(new FaceFeatureTranslator())
.optEngine("PyTorch") // Use PyTorch engine
.optDevice(Device.cpu())
// .optOption("mapLocation", "true")
// .optDevice(Device.cpu())
.optProgress(new ProgressBar())
.build();
return criteria;

2
7_engine_hub/voiceprint_search/voiceprint-search/src/main/java/me/aias/common/voice/VoiceprintModel.java
View File

@ -39,7 +39,7 @@ public final class VoiceprintModel {
.optModelUrls(modelUri)
.optTranslator(new VoiceprintTranslator())
.optEngine("PaddlePaddle") // Use PyTorch engine
//.optOption("mapLocation", "true")
//.optDevice(Device.cpu())
.optProgress(new ProgressBar())
.build();
return criteria;

2
8_suite_hub/face_search/face_search/src/main/java/me/aias/common/face/FaceDetectionModel.java
View File

@ -46,7 +46,7 @@ public final class FaceDetectionModel {
.optModelUrls(layoutUri)
.optTranslator(translator)
.optEngine("PyTorch") // Use PyTorch engine
.optOption("mapLocation", "true")
.optDevice(Device.cpu())
.optProgress(new ProgressBar())
.build();
return criteria;

2
8_suite_hub/face_search/face_search/src/main/java/me/aias/common/face/FaceFeatureModel.java
View File

@ -38,7 +38,7 @@ public final class FaceFeatureModel {
.optModelUrls(layoutUri)
.optTranslator(new FaceFeatureTranslator())
.optEngine("PyTorch") // Use PyTorch engine
.optOption("mapLocation", "true")
.optDevice(Device.cpu())
.optProgress(new ProgressBar())
.build();
return criteria;

2
8_suite_hub/question_answering_system/qa-system/src/main/java/me/aias/common/sentence/SentenceEncoderModel.java
View File

@ -47,7 +47,7 @@ public final class SentenceEncoderModel {
.optModelUrls(modelUri)
.optTranslator(new SentenceTransTranslator())
.optEngine("PyTorch") // Use PyTorch engine
.optOption("mapLocation", "true")
.optDevice(Device.cpu())
.optProgress(new ProgressBar())
.build();
return criteria;