Comments (1)
i would suggest loading the state dict and print the keys of the weights. Most probably since the authors used multigpu training (dataparallel module) , the pretrained weights keys have something like "module." prepended to the keys. Probably the less hacky way to do this is to use the same dataparallel module.
I think for your case, you can just load the state dict, iterate over the keys and values, and write another state dict with the corrected keys. THen you load the corrected state dict. This is a bit hacky but it worked for me.
from yowo.
Related Issues (20)
- Working on dataset other than humans HOT 3
- Can you provide each type of map in ava
- About the envs setup HOT 2
- KeyError: 'exp_avg' HOT 1
- YOWO not using GPUs? HOT 1
- model is None
- how is the yolo.weights trained?
- Training YOWO on a custom dataset HOT 1
- Dropbox links for pre-trained models for J-HMDB, UCF and their annotations give 404 error HOT 1
- The [email protected] results reported in the paper are for validation or test splits of the AVA dataset?
- A stronger YOWO achieved by us. HOT 4
- not find yowo_jhmdb21_32f_best.pth
- plz,how to make and train my own dataset?
- test_video_ava.py error
- ava_detection_val_boxes_and_labels.csv is missing
- ava_classnames.json is missing
- /usr/home/sut path HOT 1
- animal action reconginition
- Training YOWO on a customized dataset HOT 2
- This code lacks any Conda environment or usage instructions.
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from yowo.