switchablenorms / DeepFashion2

DeepFashion2 Dataset https://arxiv.org/pdf/1901.07973.pdf

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

How to train with entire dataset? Anyone mind sharing their trained model?

lolitsgab opened this issue · comments

Running on Google Colab and training using Mask-RCNN. I converted the dataset into the COCO format using the given script, but it seem's that it can only load about 32k images into Mask RCNN before I run out of memory (I mainly suspect pycoco's loadAnns()).

I've been training with 32k images at a time (complete training with 32k images, retrain with 32k images part 2, retrain with other 32k images... etc). So far I've only done the first three 32k image batches, and it is obviously very time consuming as I have to keep retraining. Is there a way to circumvent this? Anyone mind sharing their trained model?

i have trained 191961 train images and 32153 validation images but accuracy not good (39 epoch)