Closed adammenges closed 5 years ago
the problem is that you are trying to load the backend weigths into whole model, it is not possible
Apologies, I don't quite understand. The backend_weights there are the pertained weights I downloaded from this repo. I put them in backend_weights
folder after trying them at the root (like the README mentioned), with the same error. Are you saying there's something wrong in my config file?
just put the weights file in your root folder and don't specify any weights in the config file, the pre trained weights for the backend will be loaded automatically
When I do that I see:
⇒ python3 train.py -c config.json
Using TensorFlow backend.
2019-06-18 12:17:43.120868: I tensorflow/core/platform/cpu_feature_guard.cc:141] Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2 AVX512F FMA
Insert a comment about this training: test3
Redirecting ./racconSqueezeNet.h5 file name to ../backupSqueezeNet/SqueezeNet_VOC_20190618121743/SqueezeNet_VOC_20190618121743.h5.
Redirecting ./logsSqueezeNet/example tensorborad log to ../backupSqueezeNet/SqueezeNet_VOC_20190618121743/logs.
100%|██████████████████████████████████████████████████████████████████████████████████████████████| 190/190 [00:00<00:00, 6296.01it/s]
100%|████████████████████████████████████████████████████████████████████████████████████████████████| 10/10 [00:00<00:00, 7434.07it/s]
Seen labels: {'raccoon': 207}
Given labels: ['raccoon']
Overlap labels: {'raccoon'}
WARNING:tensorflow:From /usr/local/lib/python3.7/site-packages/tensorflow/python/framework/op_def_library.py:263: colocate_with (from tensorflow.python.framework.ops) is deprecated and will be removed in a future version.
Instructions for updating:
Colocations handled automatically by placer.
Unable to load backend weights. Using a fresh model
Specifically: Unable to load backend weights. Using a fresh model
Does that just mean it's the first training run, so there's no previous model to load, but that the pertained weights have been picked up? I don't see anything in the logs about pertained weights.
In this yolo we have two types of pre treined weights, the first one is the weights for the backend model, suck as Full Yolo, Mobilenet and so on, the second case is the weights for the whole model, where you can set into the config file, the whole model weights overwrite the backend weights when it is loaded.
create a folder called "backend_weights" inside your root directory and put the mobile weights there. you can see it here.
Ah, sounds good. Thanks!
Starting with the mobilenet backend and pertained weights, it errors with:
Config: