423A35C7/pytorch3d
1
0
Fork 0
mirror of https://github.com/facebookresearch/pytorch3d.git synced 2025-08-02 03:42:50 +08:00
Code Issues Packages Projects Releases Wiki Activity

make lpips optional

Browse Source 
Summary: So that lpips isn't needed in readthedocs

Reviewed By: shapovalov

Differential Revision: D40859351

fbshipit-source-id: 58933bd3b71e78e658fbce56c3663b11ee2331f6
This commit is contained in:
Jeremy Reizenstein 2022-10-31 14:44:41 -07:00 committed by Facebook GitHub Bot
parent eff0aad15a
commit 322c8f2f50
1 changed files with 11 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
pytorch3d/implicitron/evaluation/evaluator.py
View File

@ -5,13 +5,12 @@
# LICENSE file in the root directory of this source tree. # LICENSE file in the root directory of this source tree.
import copy import copy
import json import json
import logging import logging
import os import os
import warnings
from typing import Any, Dict, List, Optional, Tuple from typing import Any, Dict, List, Optional, Tuple
import lpips
import torch import torch
import tqdm import tqdm
@ -89,8 +88,16 @@ class ImplicitronEvaluator(EvaluatorBase):
Returns: Returns:
A dictionary of results. A dictionary of results.
""" """
lpips_model = lpips.LPIPS(net="vgg") try:
lpips_model = lpips_model.to(device) import lpips
lpips_model = lpips.LPIPS(net="vgg")
lpips_model = lpips_model.to(device)
except ImportError:
warnings.warn(
"lpips library NOT FOUND. lpips losses will not be calculated"
)
lpips_model = None
model.eval() model.eval()