fix bugs

taais-common/taais-common-core/src/main/java/com/taais/common/core/utils/file/MimeTypeUtils.java

@@ -37,7 +37,7 @@ public class MimeTypeUtils {
         // pdf
         "pdf",
         // pt
-        "pt"
+        "pt", "pth"
     };
 
     public static String getExtension(String prefix) {

taais-modules/taais-biz/src/main/java/com/taais/biz/service/impl/DataAugmentationServiceImpl.java

@@ -254,7 +254,6 @@ public class DataAugmentationServiceImpl extends BaseServiceImpl<DataAugmentatio
         makeDir(outputPath.toString());
         makeDir(logPath.toString());
 
-        System.out.println("inputPath:" + inputPath.toString());
         ZipUtils.unzip(resourcePath, inputPath.toString());
 //        //获取inputPath的下一级目录
 //        if (Files.exists(inputPath) && Files.isDirectory(inputPath)) {

taais-modules/taais-biz/src/main/java/com/taais/biz/service/impl/DataServiceImpl.java

@@ -67,7 +67,7 @@ public class DataServiceImpl extends BaseServiceImpl<DataMapper, Data> implement
     private static final String ZIP = ".zip";
     private static final String RAR = ".rar";
     private static final String TXT = ".txt";
-    private static final String[] VALID_EXTENSIONS = {".jpg", ".jpeg", ".png"};
+    private static final String[] VALID_EXTENSIONS = {".jpg", ".jpeg", ".png", ".bmp"};
     private static final String PYTHON_DATA_AMPLIFY_API = "http://127.0.0.1:9096/api/data/amplify";
     private static final String RESULT_CODE = "status";
     private static final String RESULT_STATUS = "200";

taais-modules/taais-biz/src/main/java/com/taais/biz/service/impl/TargetIdentificationSubtaskServiceImpl.java

@@ -246,8 +246,7 @@ public class TargetIdentificationSubtaskServiceImpl extends BaseServiceImpl<Targ
                 AlgorithmModelVo bo = algorithmModelService.selectById(Long.valueOf(_modelId));
                 String path = bo.getModelAddress().replace("/profile", BizConstant.DOCKER_BASE_PATH);
                 algorithmRequestDto.getOtherParams().put("pretrained", true);
-                // todo pretrained_model is a number
-//                algorithmRequestDto.getOtherParams().put("pretrained_model", path);
+                algorithmRequestDto.getOtherParams().put("pretrained_model", path);
             }
         } else if (taskName.contains("验证")) {
             String[] urls = url.split(";;;");