lucasjinreal / nb

Neural Network Blocks - Collect all kinds of fancy model blocks for you to build more powerful neural network model.
Other
238 stars 53 forks source link

Feature extraction of an image using back-bone of yolov5 #11

Closed pgupta119 closed 2 years ago

pgupta119 commented 2 years ago

Hi, I am using driving data and I would to get features from the backbone of the yolov5. How Could I get the output from the back bone?

Thanks