Commit 1e309842 authored by BO ZHANG's avatar BO ZHANG 🏀
Browse files

set binary=True explicitly

parent 34f9bb57
...@@ -100,11 +100,11 @@ class CsstMscInstrumentProc(CsstProcessor): ...@@ -100,11 +100,11 @@ class CsstMscInstrumentProc(CsstProcessor):
model = deepCR(clean_model, inpaint_model, device='CPU', hidden=50) model = deepCR(clean_model, inpaint_model, device='CPU', hidden=50)
if self.n_jobs > 1: if self.n_jobs > 1:
masked, cleaned = model.clean( masked, cleaned = model.clean(
self.__img, threshold=0.5, inpaint=True, segment=True, patch=256, parallel=True, self.__img, threshold=0.5, inpaint=True, binary=True, segment=True, patch=256, parallel=True,
n_jobs=self.n_jobs) n_jobs=self.n_jobs)
else: else:
masked, cleaned = model.clean( masked, cleaned = model.clean(
self.__img, threshold=0.5, inpaint=True, segment=True, patch=256, parallel=False, self.__img, threshold=0.5, inpaint=True, binary=True, segment=True, patch=256, parallel=False,
n_jobs=self.n_jobs) n_jobs=self.n_jobs)
else: else:
cleaned, masked = cosmicray_lacosmic(ccd=self.__img, cleaned, masked = cosmicray_lacosmic(ccd=self.__img,
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment