search

fanq15 / FewX

FewX is an open-source toolbox on top of Detectron2 for data-limited instance-level recognition tasks.
https://github.com/fanq15/FewX
MIT License
346 stars 48 forks source link

Where can I find the model_final.pth? #20

Open YuxuejieYOKI opened 3 years ago

YuxuejieYOKI commented 3 years ago

Thanks for your wonderful work! I'm wondering how can I get the pretrained model parameters dictionary 'model_final.pth' mentioned in your code. It's 'pth' not 'pkl'. I have downloaded the 'model_final.pkl' from the url of readme.md. Does the 'pth' file is required to train and save by myself? Would you please to share the best 'pth' file? Thanks again!

Yu

fanq15 commented 3 years ago

Could you tell me the code line?

YuxuejieYOKI commented 3 years ago

Thanks for your attention. I have got the 'model_final.pth' after training. However, I use single-machine-single-gpu and the result is not so satisfactory. Here I show the log.txt and some prediction results demo in the appendix. log.txt img139 img285 img632

Could you please give some suggestions? Thanks a lot! ~@_/@~ The AP metric results for coco dataset are as follows:

[11/23 17:06:42] d2.engine.defaults INFO: Evaluation results for coco_2017_val in csv format: [11/23 17:06:42] d2.evaluation.testing INFO: copypaste: Task: bbox [11/23 17:06:42] d2.evaluation.testing INFO: copypaste: AP,AP50,AP75,APs,APm,APl [11/23 17:06:42] d2.evaluation.testing INFO: copypaste: 1.1462,2.5333,0.8975,0.6895,1.4582,1.7250

gmkproofing commented 2 years ago

Thanks for your attention. I have got the 'model_final.pth' after training. However, I use single-machine-single-gpu and the result is not so satisfactory. Here I show the log.txt and some prediction results demo in the appendix. log.txt img139 img285 img632

Could you please give some suggestions? Thanks a lot! ~@_/@~

The AP metric results for coco dataset are as follows: [11/23 17:06:42] d2.engine.defaults INFO: Evaluation results for coco_2017_val in csv format: [11/23 17:06:42] d2.evaluation.testing INFO: copypaste: Task: bbox [11/23 17:06:42] d2.evaluation.testing INFO: copypaste: AP,AP50,AP75,APs,APm,APl [11/23 17:06:42] d2.evaluation.testing INFO: copypaste: 1.1462,2.5333,0.8975,0.6895,1.4582,1.7250

Hi, how did you get the model_final.pth? When I sh make.sh, it says that :"AssertionError: Checkpoint ./output/fsod/finetune_dir/R_50_C4_1x/model_final.pth not found!", thank you very much !