|  |  | 
					
						
						|  | inp_root=r"白鹭霜华长条" | 
					
						
						|  | opt_root=r"opt" | 
					
						
						|  | f0_up_key=0 | 
					
						
						|  | person=r"weights\洛天依v3.pt" | 
					
						
						|  |  | 
					
						
						|  | device = "cuda:0" | 
					
						
						|  | is_half=True | 
					
						
						|  | n_cpu=0 | 
					
						
						|  |  | 
					
						
						|  | import torch | 
					
						
						|  | if(torch.cuda.is_available()==False): | 
					
						
						|  | print("没有发现支持的N卡,使用CPU进行推理") | 
					
						
						|  | device="cpu" | 
					
						
						|  | is_half=False | 
					
						
						|  | if(device!="cpu"): | 
					
						
						|  | gpu_name=torch.cuda.get_device_name(int(device.split(":")[-1])) | 
					
						
						|  | if("16"in gpu_name): | 
					
						
						|  | print("16系显卡强制单精度") | 
					
						
						|  | is_half=False | 
					
						
						|  | from multiprocessing import cpu_count | 
					
						
						|  | if(n_cpu==0):n_cpu=cpu_count() | 
					
						
						|  | if(is_half==True): | 
					
						
						|  |  | 
					
						
						|  | x_pad       =   3 | 
					
						
						|  | x_query     =   10 | 
					
						
						|  | x_center    =   60 | 
					
						
						|  | x_max       =   65 | 
					
						
						|  | else: | 
					
						
						|  |  | 
					
						
						|  | x_pad       =   1 | 
					
						
						|  |  | 
					
						
						|  |  | 
					
						
						|  |  | 
					
						
						|  |  | 
					
						
						|  | x_query     =   6 | 
					
						
						|  | x_center    =   38 | 
					
						
						|  | x_max       =   41 |