MBLLEN.pytorch

MBLLEN.pytorch copied to clipboard

region loss存在错误

2

value = -torch.topk(-gray, int(gray.shape[0] * 0.4))[0][0] 您好，您这里求得了gray的最小值，而不是40%所在的值。 举例如下图所示：  我觉得应该是这样写可能更符合论文中描述的前40%最暗的像素值： 

linklist2

(mbllentorch) E:\MBLLEN.pytorch\MBLLEN.pytorch-master>python infer.py Traceback (most recent call last): File "infer.py", line 5, in from main import Model File "E:\MBLLEN.pytorch\MBLLEN.pytorch-master\main.py", line 6, in from pytorch_lightning import LightningModule, LightningDataModule, Trainer File "D:\anaconda\envs\mbllentorch\lib\site-packages\pytorch_lightning\__init__.py",...

rebornone1

请问是否有pytorch 的视频实现呢

yejing0