StevenLiuWen / ano_pred_cvpr2018

Official implementation of Paper Future Frame Prediction for Anomaly Detection -- A New Baseline, CVPR 2018
433 stars 126 forks source link

训练自己的数据集,损失一直增长 #50

Closed nobody-cheng closed 2 years ago

nobody-cheng commented 4 years ago

你好 我拿了17个正常的地铁(包括高峰期和正常人流)视频,数据集均是按照跳2帧切一张图,17个文件夹接近20w的图片库,按照说明进行一些参数设置 --iters = 80000 --batch = 16 损失图如下,adversarial loss和train_psnr_error generator loss均是越来越大 Snipaste_2020-06-19_18-17-48 Snipaste_2020-06-19_18-17-58

zachluo commented 4 years ago

@nobody-cheng train_psnr_error越高越好,generator loss越低越好,你这不是很正常吗?adversarial loss变大很正常,可以多训练。