Apppublic

procgne/Plonk

sourceHugging Facemitupdated 10mo agoView on Hugging Face

0likes

rootmodels__pycache__postprocessing.cpython-310.pyc

postprocessing.cpython-310.pyc10 linesDownload Raw Back to __pycache__

1o

2�q�f��@sDddlmZddlZddlZGdd�dej�ZGdd�dej�ZdS)�Nc�$eZdZ�fdd�Zdd�Z�ZS)�UnormGPScs2t���|�dt�tjdtjg��d��dS)N�
gps_normalizeg�?r)�super�__init__�register_buffer�torch�Tensor�np�pi�	unsqueeze��self��	__class__��:/home/dufour/Documents/diff_plonk/models/postprocessing.pyrs3(zUnormGPS.__init__cCst�|dd�}||jS)z0Unormalize latitude longtitude radians to -1, 1.������)r�clampr)r�xrrr�forward4s5zUnormGPS.forward��__name__�6__module__�__qualname__rr�
__classcell__rrrrrsrcr)�CartesiantoGPScst���dS)N)rrr
rrrrszCartesiantoGPS.__init__cCsT|dd�df}|dd�df}|dd�df}|��}|�|�}tj||gdd�S)Nrr�r)�dim)�arcsin�atan2r�stack)r�	cartesianr�y�z�lat�lonrrrrs7zCartesiantoGPS.forwardrrrrrrsr)�torch.nn�nnr�numpyr8�Modulerrrrrr�<module>s910