Aboud black mask image when process local custom data #561
Replies: 3 comments 2 replies
-
Hi, I have downloaded the contents of the Dropbox link and extracted it in a directory named weights/ I am getting the following error: File "/home/imerit/.local/lib/python3.8/site-packages/tensorflow/python/training/saver.py", line 1290, in restore Can you kindly help me out? |
Beta Was this translation helpful? Give feedback.
-
@arijit-imerit --weights ./weights/xxxx.ckpt |
Beta Was this translation helpful? Give feedback.
-
@arijit-imerit You may google how to restore tensorflow model from checkpoint file:) |
Beta Was this translation helpful? Give feedback.
-
Lots of issues mentioned about the black mask image problem during process their own custorm data. The problem is mainly caused by the cluster postprocess strategy. Here is a sample collected in the closed issues(#382). User have found black mask image when using pretrained model to inference his own image data. The test results are listed here.
mask_image_before_adjust
You may find the instance segmentation result may somehow identify different lanes although it was not that good here but it's a another question. Why would the mask image still become empty? The reason was the dbscan cluster params was not proper configured for that data. When I enlarge the dbscan eps param from 0.35 to 0.5 and reduce min_samples params from 1000 to 250. The cluster could work well on this data. The test was listed here.
mask_image_after_adjust
So first thing you may need to do when faced with such a problem is to adjust the dbscan params to see if it helps. But if you can not even identify different lanes on instance segmentation result or binary segmentation result what you need is to retrain the model rather than adjust some of the params.
Anyway hope it helps you guys !!!
Beta Was this translation helpful? Give feedback.
All reactions