Comments (5)
Hi,
Thanks for reading.
We provide pretrained CSRA model like ResNet-101, including its backbone weights and classifier weights. For feature extraction, you can only load backbone weights of our pretrained model. The following two command might help
csra_model_weights = torch.load("CSRA_MODEL.pth")
print(csra_model_weights.keys())
then the keys with "classifier" in it can be safely discarded and the left keys and its weights are backbone parameters for feature extraction.
Best,
Ke
from csra.
Hi Mr. Ke,
thank you so much for your detailed reply. I'll look deep into it.
Best regards,
Hui
from csra.
Hi Mr. Ke,
I have a question regarding loading the model. Do I have to delete classifier part of your module before training? Or just not load the weights of classifier and train as a whole? Thank you in advance.
Best regards,
Hui
from csra.
Hi, Hui
Usually the number of class in your own dataset can be different from VOC, MS-COCO, so the pretrained classifier's weights can not be loaded due to different num-class.
1st, load only the backbone's weights.
2nd, initialize the new classifier's weights (using the number of class in your dataset).
Then you can train as a whole using either your our method or our CSRA method.
Best,
Ke
from csra.
Hi Mr. Ke,
thank you again.
Best regards,
Hui
from csra.
Related Issues (20)
- Is the code consistent with the description in the paper? HOT 4
- MobileNet; input size HOT 2
- Transformer on WiderAttribute predicition HOT 1
- Is 'normalized' classifier necessary? HOT 3
- some questions about val.py HOT 1
- The problems of val.py? HOT 1
- Partial-label HOT 1
- About the pretrained vit. HOT 2
- Question about Attention Image or Heatmap Generation HOT 1
- Specific improvements to vit HOT 3
- Some problems about vision transformer HOT 1
- Details about baseline resnet-101 in paper HOT 1
- Load Model İssue? HOT 1
- Cross Validation HOT 2
- MobileNet implementation of CSRA HOT 4
- About wider_attribute is Cropping individual person attributes ?
- How to use VIT-224 pretrain weight with 448 input size
- Use CSRA module in the ResNet50 model
- visualize function
- Global feature vector
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 csra.