menyifang / DCT-Net

Official implementation of "DCT-Net: Domain-Calibrated Translation for Portrait Stylization", SIGGRAPH 2022 (TOG); Multi-style cartoonization
Apache License 2.0
734 stars 74 forks source link

CPU模式运行正常。 GPU的时候就报错 。 #75

Open VictorQiuYY opened 1 month ago

VictorQiuYY commented 1 month ago

pciBusID: 0000:01:00.0 2024-06-06 19:09:39.120194: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cudart64_100.dll 2024-06-06 19:09:39.120241: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cublas64_100.dll 2024-06-06 19:09:39.120272: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cufft64_100.dll 2024-06-06 19:09:39.120301: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library curand64_100.dll 2024-06-06 19:09:39.120330: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cusolver64_100.dll 2024-06-06 19:09:39.120352: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cusparse64_100.dll 2024-06-06 19:09:39.120374: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cudnn64_7.dll 2024-06-06 19:09:39.120412: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1746] Adding visible gpu devices: 0 2024-06-06 19:09:39.120457: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1159] Device interconnect StreamExecutor with strength 1 edge matrix: 2024-06-06 19:09:39.120478: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1165] 0 2024-06-06 19:09:39.120498: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1178] 0: N 2024-06-06 19:09:39.120589: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1304] Created TensorFlow device (/device:GPU:0 with 4912 MB memory) -> physical GPU (device: 0, name: NVIDIA GeForce RTX 4090, pci bus id: 0000:01:00.0, compute capability: 8.9) 9it [02:54, 19.37s/it] Traceback (most recent call last): File "run_vid.py", line 42, in process(args) File "run_vid.py", line 26, in process result = img_cartoon(img[..., ::-1]) File "C:\ProgramData\anaconda3\envs\dctnet\lib\site-packages\modelscope\pipelines\base.py", line 196, in call output = self._process_single(input, args, kwargs) File "C:\ProgramData\anaconda3\envs\dctnet\lib\site-packages\modelscope\pipelines\base.py", line 233, in _process_single out = self.forward(out, forward_params) File "C:\ProgramData\anaconda3\envs\dctnet\lib\site-packages\modelscope\pipelines\cv\image_cartoon_pipeline.py", line 102, in forward landmarks = self.detect_face(img) File "C:\ProgramData\anaconda3\envs\dctnet\lib\site-packages\modelscope\pipelines\cv\image_cartoon_pipeline.py", line 79, in detectface boxes, landmarks, = self.facer.run(img) File "C:\ProgramData\anaconda3\envs\dctnet\lib\site-packages\modelscope\models\cv\cartoon\facelib\facer.py", line 41, in run landmarks, states = self.face_landmark(image, boxes) File "C:\ProgramData\anaconda3\envs\dctnet\lib\site-packages\modelscope\models\cv\cartoon\facelib\face_landmark.py", line 40, in call landmark, state = self._one_shot_run(img, bbox, i) File "C:\ProgramData\anaconda3\envs\dctnet\lib\site-packages\modelscope\models\cv\cartoon\facelib\face_landmark.py", line 103, in _one_shot_run int(x_y[1] cfg.KEYPOINTS.input_shape[1] + bbox[1] - add) OverflowError: cannot convert float infinity to integer

GPU模式下在 2024-06-06 19:09:39.120374: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cudnn64_7.dll 时会卡很久 。 然后报错