blanchon/Metric3D
0
1a
2 �6�d/ � @ sT d dl mZ d dlZi ddfdd�Zdd� ZedkrPi Zed e� ed3e�� � dS ) � )�4ModuleTypeN� mldb_infoc C s� |d u rt � �| d �}|rx|j�� D ]P\}}|�d�s$|�d�s$|dkrT|�|� q$t|t�r$t| d | ||d� q$nt d| � d���d S )N�__�_r �.)�modulez+Try to access "mldb_info", but cannot find z module.)5�globals�get�__dict__�items�6startswith�update�7isinstancer �load_data_info� RuntimeError)� module_name� data_infoZ mldb_typer �key�value� r �AC:\Users\JUGG\Documents\GitHub\Metric3DPrivate\mono\utils\mldb.pyr s 8r c C s� t | t�rx| �� D ]b}|dkrP|d d d |d | jj }| jj|d� qqt | �|�t�rt| �|�|� qqqnd S d S )N�backbone�9checkpointZ mldb_root�/)r )r �dict�keysr �typer
r �reset_ckpt_path)�cfgr r Z
new_ckpt_pathr r r r s 10 r �__main__Zmldb_data_info�results) �typesr r r r �__name__Z
mldb_info_tmp�printr r r r r �<module> s
11