Running the code runs into some problems

[hexiao0275]

Thank you for your awesome work! What files are contained in the datasets_dir = "/lustre/groups/aih/raheleh.salehi/Aug_features_datasets/" in the Dataloader.py file? I am unable to run train.py after running Mask RCNN.

[RahelehSalehi]

Hi @hexiao-cs and sorry to respond so late. I would be happy to know how our method performs on your data.
This path directory is for the feature vector obtained from Mask R-CNN with a shape of 256x14x14.
Please check out the Mask R-CNN folder for more details about getting the feature vectors.
Based on how your data is stored, you need to rewrite the init function, I wrote init function such that it loads everything into memory first.
Best,
Raheleh