GitHub - khanhnguyen-96/license_plate_super

Branches Tags

Name		Name	Last commit message	Last commit date
Latest commit History 82 Commits
.vscode		.vscode
L1CLoss		L1CLoss
ML1Loss		ML1Loss
data		data
digits		digits
experiments/RDN		experiments/RDN
loss		loss
model		model
pytorch_ssim		pytorch_ssim
tools		tools
.gitignore		.gitignore
.plot_multi_v1.py		.plot_multi_v1.py
.psnrssim_DDBPN.sh		.psnrssim_DDBPN.sh
.psnrssim_EDSR.sh		.psnrssim_EDSR.sh
.psnrssim_RDN.sh		.psnrssim_RDN.sh
.psnrssim_v0.1.py		.psnrssim_v0.1.py
.psnrssim_v1.2.1.py		.psnrssim_v1.2.1.py
.psnrssim_v1.2.py		.psnrssim_v1.2.py
.psnrssim_v1.py		.psnrssim_v1.py
.trainer_13.10.2018.py		.trainer_13.10.2018.py
.trainer_18.11.2018.py		.trainer_18.11.2018.py
.trainer_24.11.2018.py		.trainer_24.11.2018.py
.trainer_5.11.2018.py		.trainer_5.11.2018.py
.trainer_SSIM.py		.trainer_SSIM.py
README.txt		README.txt
__init__.py		__init__.py
dataloader.py		dataloader.py
dataloader_old.py		dataloader_old.py
demo.sh		demo.sh
eval.sh		eval.sh
evalrect.sh		evalrect.sh
evalsquare.sh		evalsquare.sh
lpt_utils.py		lpt_utils.py
main.py		main.py
missclass.py		missclass.py
ml1_test.py		ml1_test.py
option.py		option.py
plateLicenseSR.ipynb		plateLicenseSR.ipynb
plm.sh		plm.sh
plot_multi.py		plot_multi.py
plot_multi_v2.py		plot_multi_v2.py
psnrssim.py		psnrssim.py
psnrssim_v1.2.2.py		psnrssim_v1.2.2.py
psnrssim_v1.2.3.py		psnrssim_v1.2.3.py
resize.py		resize.py
setup.cfg		setup.cfg
square_to_rect.py		square_to_rect.py
sr.py		sr.py
srserver.py		srserver.py
srserver2.py		srserver2.py
str_sim.py		str_sim.py
template.py		template.py
testssim.py		testssim.py
trainer.py		trainer.py
traintest.sh		traintest.sh
utility.py		utility.py
xosquare.sh		xosquare.sh

Repository files navigation

Đây là thư mục chứa code.
Script để train và test: traintest.sh
Để train có thể dùng lệnh: nohup ./traintest.sh trainrdnsq &
Để test dùng lệnh: ./traintest.sh testrdnsq {path_to_model} [path_to_testset]. path_to_testset chứa 2 thư mục LR/ và HRS/, LR/ chứa ảnh để test, HRS/ chứa ảnh để so sánh. Ví dụ: ./traintest.sh testrdnsq ../experiment/RDN_D16C8G64_square23/model/model_best.pt
Ảnh khôi phục được lưu tại ../experiment/test/results/