From 802a7f10e92237e30ae85df2e4151f5fd39f4e61 Mon Sep 17 00:00:00 2001 From: kotauchisunsun Date: Fri, 28 Feb 2020 00:45:34 +0900 Subject: [PATCH] run on cpu --- apps/eval.py | 4 ++-- requirements.txt | 23 +++++++++++++++++++++++ 2 files changed, 25 insertions(+), 2 deletions(-) create mode 100644 requirements.txt diff --git a/apps/eval.py b/apps/eval.py index ff144ab4b..5800ecd9f 100755 --- a/apps/eval.py +++ b/apps/eval.py @@ -34,7 +34,7 @@ def __init__(self, opt, projection_mode='orthogonal'): transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5)) ]) # set cuda - cuda = torch.device('cuda:%d' % opt.gpu_id) + cuda = torch.device('cuda:%d' % opt.gpu_id) if torch.cuda.is_available() else torch.device('cpu') # create net netG = HGPIFuNet(opt, projection_mode).to(device=cuda) @@ -120,4 +120,4 @@ def eval(self, data, use_octree=False): data = evaluator.load_image(image_path, mask_path) evaluator.eval(data, True) except Exception as e: - print("error:", e.args) \ No newline at end of file + print("error:", e.args) diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 000000000..bdbd6693b --- /dev/null +++ b/requirements.txt @@ -0,0 +1,23 @@ +cycler==0.10.0 +decorator==4.4.1 +imageio==2.8.0 +kiwisolver==1.1.0 +matplotlib==3.1.3 +networkx==2.4 +numpy==1.18.1 +opencv-python==4.2.0.32 +pathlib==1.0.1 +Pillow==7.0.0 +PyOpenGL==3.1.5 +pyparsing==2.4.6 +python-dateutil==2.8.1 +PyWavelets==1.1.1 +scikit-image==0.16.2 +scipy==1.4.1 +Shapely==1.7.0 +six==1.14.0 +torch==1.4.0 +torchvision==0.5.0 +tqdm==4.43.0 +trimesh==3.5.23 +xxhash==1.4.3