search

ISICV / ManTraNet

ManTra-Net: Manipulation Tracing Network For Detection And Localization of Image Forgeries With Anomalous Features
269 stars 70 forks source link

OOM error #5

Closed zz397 closed 3 years ago

zz397 commented 4 years ago

When I tried image of size (1000,2000), I got OOM error with a 12gb memory GPU. Can you suggest how to proceed? Should I resize it smaller?

rex-yue-wu commented 3 years ago

Three solutions:

  1. resize it to a smaller size
  2. crop images to overlapping smaller blocks and process one at a time
  3. divide the original network into small subnetworks, and run one network at each time