From b7cb9f2877b3a52e29c21ef2b0a6000779136e74 Mon Sep 17 00:00:00 2001 From: Ameya Daigavane Date: Wed, 16 Oct 2019 18:18:05 +0000 Subject: [PATCH 1/2] Fix installation. --- README.md | 3 ++- ext/setup.py | 4 ++-- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/README.md b/README.md index 2a90df5..5130258 100644 --- a/README.md +++ b/README.md @@ -14,7 +14,8 @@ The implementation is based on the architectures of [DeepLabV3+](https://github. ### Gaussian PSF Layer In order to build the Gaussian PSF layer on your own machine, run the following line ``` -python ext/setup.py install +cd ext +python setup.py install ``` ## Datasets diff --git a/ext/setup.py b/ext/setup.py index 81e774d..b2d084a 100644 --- a/ext/setup.py +++ b/ext/setup.py @@ -6,9 +6,9 @@ ext_modules=[ CUDAExtension('gauss_psf_cuda', [ 'gauss_psf_cuda.cpp', - 'gauss_psf_kernel.cu', + 'gauss_psf_cuda_kernel.cu', ]), ], cmdclass={ 'build_ext': BuildExtension - }) \ No newline at end of file + }) From 6481c11e79ebac5b82c8f9ed5c3db105616e0f73 Mon Sep 17 00:00:00 2001 From: Ameya Daigavane Date: Wed, 16 Oct 2019 18:37:21 +0000 Subject: [PATCH 2/2] Fix gauss_psf_cuda import. --- Utils/GaussPSFLayer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Utils/GaussPSFLayer.py b/Utils/GaussPSFLayer.py index 5f976a4..8878adf 100644 --- a/Utils/GaussPSFLayer.py +++ b/Utils/GaussPSFLayer.py @@ -2,7 +2,7 @@ import torch.nn as nn from torch.autograd import Function -import gauss_psf +import gauss_psf_cuda class GaussPSFFunction(Function):