CAT-Net icon indicating copy to clipboard operation
CAT-Net copied to clipboard

dataset processing

Open dan326326 opened this issue 3 years ago • 1 comments
trafficstars

Hello,Thank you for your wonderful code!I see COLUMB data set is used in the paper, may I ask how do you deal with the mask of this data set

dan326326 avatar Dec 03 '21 06:12 dan326326

Hi, You may use the masks in the edgemask directory in 4cam_splc with a proper threshold (like 120/255).

I've done this by:

mask = np.array(Image.open(mask_path))[:,:,1]
mask[mask <= 120] = 0
mask[mask > 0] = 1

CauchyComplete avatar Dec 06 '21 04:12 CauchyComplete