Open lucky-starh opened 4 years ago
Thank you. I have one more question: OutOfMemoryError. The defualt batch_size =20, can only change the bitch_size ?
@lucky-starh if the input image is 512 x 512, batch size =20 is too large to the GPU memory, batch size = 5 will be better.
Has been an error: Crop image error, always appear in the Img_RightCamera/SN029 or Img_RightCamera/SN029 folder
@lucky-starh Maybe the head is not frontal position, you need to check that reason. But I remember I have already added the exception code for handling this case....
I haven't found the conversion to BGR->RGB image in get_example() function.
@lucky-starh I think you do not need to convert from BGR -> RGB. Have you fixed this issue?
If you use
memcached
andpylibmc
to store the bounding box coordinates, the training will be accelarate by the cache. You must assure the cache is correctly used in the program. When I use the Nvidia Tesla P100 card, the training will be finished in 4 days at most.