search

zhanglichao / generatedTIR_tracking

Synthetic data generation for end-to-end TIR tracking (TIP2018)
53 stars 12 forks source link

pix2pix reports an error #6

Closed ghost closed 4 years ago

ghost commented 4 years ago

Hi, thank you for sharing. I downloaded your weight file. But I encountered a problem using https://github.com/junyanz/pytorch-CycleGAN-and-pix2pix. CycleGAN can generate thermal infrared images, but pix2pix reports an error. I enter the command in the terminal as: python test.py --dataroot ./datasets/visible/ --direction AtoB --model test --name net_G_pix2pix. But the error message is AttributeError: 'Sequential' object has no attribute 'model'.

zhanglichao commented 4 years ago

Hi,

I have uploaded the GAN code for generating synthetic images, given some changes for loading the models, you could refer to the code or directly use this code.

/Lichao

ghost commented 4 years ago

Hi,

I have uploaded the GAN code for generating synthetic images, given some changes for loading the models, you could refer to the code or directly use this code.

/Lichao

Hello, I really appreciate your reply and the code Shared.I wish you more success.