search

mjkwon2021 / CAT-Net

Official code for CAT-Net: Compression Artifact Tracing Network. Image manipulation detection and localization.
210 stars 25 forks source link

Smallest Dataset size #14

Open Vadim2S opened 2 years ago

Vadim2S commented 2 years ago

I am try train on small (400 images) CASIA dataset for debug purpose.
I am get assert error: "assert 0 <= index < len(self.tamp_list), f"Index {index} is not available!"" In data_core.py I am found "self.smallest = 1869 # smallest dataset size (IMD:1869)" and presume my dataet is too small. I am wonder why smallest dataset size is exactly 1869?

CauchyComplete commented 2 years ago

You should modify self.smallest to the smallest dataset size. In your case, set it as 400. The number 1869 is the size of IMD dataset.