search

Xilinx / QNN-MO-PYNQ

BSD 3-Clause "New" or "Revised" License
242 stars 114 forks source link

training model #35

Open Changyiyu opened 5 years ago

Changyiyu commented 5 years ago

Hello, Can I get the training model of dorefanet-classifier? I want to train the plant classes.

ussamazahid96 commented 5 years ago

you should take a look at this repo

KaTaiHo commented 5 years ago

@ussamazahid96 were you able to successfully train new weights for QNN? If so, could you please share the steps to do so? I have looked at the repo that you mentioned above but couldn't get the npz files needed etc.

mohdumar644 commented 5 years ago

@KaTaiHo did you follow the steps at https://github.com/mohdumar644/QNN-MO-PYNQ/tree/master/qnn/src/training README?

If you do not use tensorpack, also see https://github.com/mohdumar644/DoReFaNet-PyTorch

KaTaiHo commented 5 years ago

@mohdumar644 Sorry I made a mistake in my previous post. I meant to ask if we needed to update the npy files such as tinier-yolo-conv0-W.npy from https://github.com/Xilinx/QNN-MO-PYNQ/blob/master/notebooks/tiny-yolo-image.ipynb.

mohdumar644 commented 5 years ago

Yes if training custom, they are needed for PS layers, only in QNN-MO-PYNQ (not BNN).

Asulr commented 4 years ago

@KaTaiHo did you follow the steps at https://github.com/mohdumar644/QNN-MO-PYNQ/tree/master/qnn/src/training README?

If you do not use tensorpack, also see https://github.com/mohdumar644/DoReFaNet-PyTorch

Hi, sorry to bother you. I have viewed the comments above and I have ran the DoreFaNet-PyTorch,and got the npz file. But when I use bin-gen from https://github.com/mohdumar644/QNN-MO-PYNQ/tree/master/qnn/src/training, I was failed. There don't have the file 0.npy.

B1D1ng commented 4 years ago

@KaTaiHo did you follow the steps at https://github.com/mohdumar644/QNN-MO-PYNQ/tree/master/qnn/src/training README? If you do not use tensorpack, also see https://github.com/mohdumar644/DoReFaNet-PyTorch

Hi, sorry to bother you. I have viewed the comments above and I have ran the DoreFaNet-PyTorch,and got the npz file. But when I use bin-gen from https://github.com/mohdumar644/QNN-MO-PYNQ/tree/master/qnn/src/training, I was failed. There don't have the file 0.npy.

Hi! sorry to bother you. Have you solved this problem and succeed training your custom dataset?

B1D1ng commented 4 years ago

@KaTaiHo did you follow the steps at https://github.com/mohdumar644/QNN-MO-PYNQ/tree/master/qnn/src/training README?

If you do not use tensorpack, also see https://github.com/mohdumar644/DoReFaNet-PyTorch

Hi, sorry to bother you. If I want to train my own custom dataset for the QNN-MO-PYNQ, I should do the steps in this README?