Modify inference.py and custom_dataset.py (just fixed some errors) b70f9e4 shgyg99 commited on May 17