tianbaochou / NasUnet

170 stars 45 forks source link

Search for a new neural architecture using my own dataset #16

Closed ghost closed 5 years ago

ghost commented 5 years ago

@tianbaochou

You have stated before that train.py uses architectures that you already searched by yourself, which you have placed in geno_searched.py

My question is: How can I search for a new neural architecture using my own dataset. I have attached a sample of the dataset below.

sample.zip

tianbaochou commented 5 years ago

@tianbaochou

You have stated before that train.py uses architectures that you already searched by yourself, which you have placed in geno_searched.py

My question is: How can I search for a new neural architecture using my own dataset. I have attached a sample of the dataset below.

sample.zip

The README file have show you how to custom your dataset. The most important thing you need do is locate the image root path. and imitate the dataset class I have overload from torch Dataset.