第一次修改:
1、 balloon.py的87、88行
# Number of training steps per epoch
STEPS_PER_EPOCH = 100
--------> 5
2、 config.py的37~49行
# Number of training steps per epoch
# This doesn't need to match the size of the training set. Tensorboard
# updates are saved at the end of each epoch, so setting this to a
# smaller number means getting more frequent TensorBoard updates.
# Validation stats are also calculated at each epoch end and they
# might take a while, so don't set this too small to avoid spending
# a lot of time on validation stats.
STEPS_PER_EPOCH = 1000
# Number of validation steps to run at the end of every training epoch.
# A bigger number improves accuracy of validation stats, but slows
# down the training.
VALIDATION_STEPS = 50
--------> 5
第一次结果: 和之前一样
第二次修改:
1、config.py的31~35行
# Number of images to train with on each GPU. A 12GB GPU can typically
# handle 2 images of 1024x1024px.
# Adjust based on your GPU memory and image sizes. Use the highest
# number that your GPU can handle for best performance.
IMAGES_PER_GPU = 2
--------> 1
第二次结果: 出现epoch1/30之后,有一段升高的波动,但又回去啦(;′⌒`)