ViolenceDetection icon indicating copy to clipboard operation
ViolenceDetection copied to clipboard

PRETRAIN_MODEL_PATH_NAME

Open heboyong opened this issue 6 years ago • 5 comments

I don not know how to set PRETRAIN_MODEL_PATH_NAME in TrainSettings.py how can I get the pretrained model,emmm or checkpoint file, I am confused

heboyong avatar Oct 19 '18 15:10 heboyong

i am having the same problem while using the pretrained model... its not picking the darknet19.pb file

mujtabaasad avatar Nov 10 '18 10:11 mujtabaasad

Hi, Sorry for the confusing variable name. The PRETRAIN_MODEL_PATH_NAME in TrainSettings.py is the path to, say if you have trained this network already, and you want to finetune again. You can set it to "" if you start from scratch. Since this project use the Darknet19 to extract the spacial feature, one should specify the path to darknet19.pb by the DARKNET19_MODEL_PATH variable in the 'src/net/*.py' (e.g. G2D19_P2OF_ResHB_1LSTM.py). Since different people may want to use different net, and some net might not want to use the Darknet19 as its base, I do not add the DARKNET19_MODEL_PATH to NetSettings.py. Sorry for the trouble.

JoshuaPiinRueyPan avatar Dec 03 '18 13:12 JoshuaPiinRueyPan

thanks for your reply,I have solved this problem after asked this question a few days,and all the model runs well,thanks again for your job

heboyong avatar Dec 03 '18 14:12 heboyong

Hey heboyong can you provide me your email address ? can you provide me your TrainSettings file?

thanks for your reply,I have solved this problem after asked this question a few days,and all the model runs well,thanks again for your job

RimshaMajeed avatar Dec 03 '18 14:12 RimshaMajeed

Hello Joshua, Can you please provide the pre-trained model and the model checkpoint for this code?

swaticolab avatar Apr 05 '19 09:04 swaticolab