You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Notice: CV based fast-search only support Android for now
Run
Before running our code, connect to your phone via USB.
If Android phone, open the USB debugging at developer options enter adb devices to ensure that the list is not empty.
If iPhone, please ensure that you have a mac. Then following this link for preparation.
It is recommended to download the pre-trained model following the link below and run the following code
Training: download and untar data into any directory, and then modify self.data_dir in those files under cnn_coarse_to_fine/data_provider directory.
Inference: download and unzip train log dirs(train_logs_coarse and train_logs_fine) into resource directory.
How to Train CNN models by yourself?
Download and untar data into any directory, and then modify self.data_dir in those files under cnn_coarse_to_fine/data_provider directory.
base.large is model dir for coarse model, base.fine is model dir for fine model, other dirs under cnn_coarse_to_fine/config are models we don't use, but if you have interests, you can try train other models by yourself.
Run python3 train.py -g 0 to train your model, -g to specify GPU to use, if you don't have GPU, training model is not recommended because training speed with CPU is very slow.
After training, move or copy .ckpt file to train log dirs(train_logs_coarse and train_logs_fine) for use.