Is it accessible to train my own dataset?

langnico / GRAINet

8 stars 4 forks source link

Is it accessible to train my own dataset? #3

Open binbinlan opened 2 years ago

binbinlan commented 2 years ago

Dear author, I am very interested in your works. After seeing your excellent results, I want to train the network on my own dataset. Can you tell me how to customize a training dataset? It mainly refers to the creation process of NPZ files, including scripts used by experts for manual annotation.

gzegers commented 1 year ago

Hi @binbinlan,

The NPZ files are exported directly from numpy arrays. You need to create an array for each necessary variable (images. histograms, tiles_names, and dm). Then you can export the NPZ using the following function:

import numpy as np np.savez_compressed('Data_V1.npz', images=images, histograms=histograms,tile_names=tile_names,dm=dm)

Best,

Gerardo