type=str,
help='Path to samples to recongize')
args = parser.parse_args()
if args.interactive and args.sample_dir:
print(
'Incompatible options. Please specify either --interactive or --sample_dir, not both at the same time'
)
exit(-1)
if not args.interactive and not args.sample_dir:
print('No action. Please specify either --interactive or --sample_dir')
exit(-1)
database = sqlite3.connect(args.database)
searcher = Searcher.load(args.index)
small_encoder = SmallEncoder().cuda()
load_model_state(args.small_encoder, small_encoder)
large_encoder = LargeEncoder().cuda()
load_model_state(args.large_encoder, large_encoder)
large_encoder.eval()
small_encoder.eval()
if args.interactive:
interactive(small_encoder, large_encoder, searcher, database,
args.augment)
exit(0)
if args.sample_dir:
process_dir(small_encoder, large_encoder, searcher, database,
