1.举例,来自知乎深度学习库的事例https://zhuanlan.zhihu.com/p/396524552
2.修改,下载之后做略微的修改指定model文件
Criteria<NDArray, Map> kpDetector =
Criteria.builder()
.setTypes(NDArray.class, Map.class)
.optTranslator(new NPtKTranslator())
.optEngine("PyTorch")
.optDevice(Device.cpu())
.optModelUrls("file:///models/my_model/1/")
.optModelName("kp_detector.pt")
.build();
Predictor<NDArray, Map> kPredictor = ModelZoo.loadModel(kpDetector).newPredictor();
//saveBoundingBoxImage(driving_video.get(0));
Criteria<List, Image> generatorCr =
Criteria.builder()
.setTypes(List.class, Image.class)
.optEngine("PyTorch")
.optTranslator(new PtGTranslator())
.optDevice(Device.cpu())
.optModelUrls("file:///models/my_model/1/")
.optModelName("generator.pt")
//.optModelPath(Paths.get("D:/work/learn/AI_MODEL/generator.pt"))
.build();
3.启动之后控制台会打印No matching cuda flavor for win found: cu65并且在线下载dll文件,如果当时网速恰好能在线下载,可忽略此文
4.解决方案:在LibUtils文件中打断点,查看dir的指定路径,没有路径自行新建,查看is地址打开后将符合后续执行逻辑的连接留下来自行下载至dir指定路径
5.完成4之后重启即可合成视频
备注:网速不好难下载的查看这个已下载的
链接:https://pan.baidu.com/s/1yaHo3tB8_CSYQbBu7WTkOA
提取码:za71
附:此文仅作为个人学习笔记,希望与各位正在学习的朋友多多交流